comment here

.vscode/settings.json

@@ -0,0 +1,20 @@
+{
+    "MicroPython.executeButton": [
+        {
+            "text": "▶",
+            "tooltip": "Run",
+            "alignment": "left",
+            "command": "extension.executeFile",
+            "priority": 3.5
+        }
+    ],
+    "MicroPython.syncButton": [
+        {
+            "text": "$(sync)",
+            "tooltip": "sync",
+            "alignment": "left",
+            "command": "extension.execute",
+            "priority": 4
+        }
+    ]
+}

src/interface.jl

@@ -37,11 +37,11 @@ function Base.getindex(container::filecontainer, idx::Int)
 
     _img = Images.load(path)
     # _img = apply(tfm, Image(_img))  # augment an image
-    img = collect(channelview(float32.(RGB.(_img))))
+    _img = channelview(float32.(RGB.(_img)))
     # img = (img .- mean(img)) ./ std(img)      # normalize the image
-    img = permutedims(img, (3, 2, 1))
-    y = container.classIdOrder[Symbol(classId)]
-    return img, y
+    img = permutedims(_img, (3, 2, 1))
+    # y = container.classIdOrder[Symbol(classId)]
+    return img
 end
 
 
@@ -104,6 +104,7 @@ end
 
 
 
+
 
 
 end # module interface