open isImage() method to package children for reuse-enabling custom d…

…atasets (#2662)
deepjavalibrary · Jun 18, 2023 · f936f40 · f936f40
1 parent 8f841e6
commit f936f40
Showing 1 changed file with 1 addition and 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
@@ -128,7 +128,7 @@ protected void listImages(Path root, List<String> 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;