-
Notifications
You must be signed in to change notification settings - Fork 138
WIP: Reading PNG images as arrays for MNIST Prediction #399
Conversation
Can you add what you made as the comment here to the |
Let's also add a test for CSVSource to make sure that the loading which was added there works. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Given yash's feedback, let's make the entrypoint
on this mnistpng
for now. So that no one gets confused. Let's leave the directory name though as png
, thanks! Looking great :)
@sakshamarora1 Looking great! Sorry to hit you with a couple more things but I just got around to responding. Thank you! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Almost there! :D Nice work!!
It looks like we also need to add to diff --git a/dffml/service/dev.py b/dffml/service/dev.py
index b88818c9..f8fa2650 100644
--- a/dffml/service/dev.py
+++ b/dffml/service/dev.py
@@ -45,6 +45,7 @@ EMAIL = config.get("user", "email", fallback="[email protected]")
CORE_PLUGINS = [
("configloader", "yaml"),
+ ("configloader", "png"),
("model", "tensorflow"),
("model", "scratch"),
("model", "scikit"), |
Add Pillow dependency
For Issue #371
Create image.csv
After training on the MNIST dataset ,
Predict command :