From 6622f211f5123b663b4104e608291556d1c9f5c4 Mon Sep 17 00:00:00 2001 From: Julian Gamble Date: Sun, 18 Jun 2023 21:29:26 +1000 Subject: [PATCH] open isImage() method to package children for reuse-enabling custom datasets --- .../djl/basicdataset/cv/classification/AbstractImageFolder.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/basicdataset/src/main/java/ai/djl/basicdataset/cv/classification/AbstractImageFolder.java b/basicdataset/src/main/java/ai/djl/basicdataset/cv/classification/AbstractImageFolder.java index 3c175f85614..bcb4c4e70ce 100644 --- a/basicdataset/src/main/java/ai/djl/basicdataset/cv/classification/AbstractImageFolder.java +++ b/basicdataset/src/main/java/ai/djl/basicdataset/cv/classification/AbstractImageFolder.java @@ -128,7 +128,7 @@ protected void listImages(Path root, List classes) { protected abstract Path getImagePath(String key); - private boolean isImage(File file) { + protected boolean isImage(File file) { String path = file.getName(); if (!file.isFile() || file.isHidden() || path.startsWith(".")) { return false;