added sample dataset

README.md

@@ -21,12 +21,12 @@ pip install access-niu
 
 ## Training
 ```bash
-python -m access_niu.train --template sample/sample_template.yml
+python -m access_niu.train --template access_niu/sample/colors/sample_template.yml
 ```
 
 ## Inference
 ```bash
-python -m access_niu.wsgi --project ./sample_project
+python -m access_niu.wsgi --project ./colors
 ```
 Now use this curl command to parse
 ```bash

access_niu/about.py

@@ -1,9 +1,9 @@
 # inspired from: https://github.com/explosion/spaCy/blob/master/spacy/about.py
 
 __title__ = "access_niu"
-__version__ = "0.0.1"
+__version__ = "0.0.1dev"
 __summary__ = "Production grade Natural Image Understanding"
-__uri__ = "https://accessai.co"
+__uri__ = "http://accessai.co"
 __author__ = "ConvexHull Technology"
 __email__ = "[email protected]"
 __license__ = "Apache 2.0"

access_niu/components/component.py

@@ -86,8 +86,8 @@ def execute(self, **kwargs):
 
         self.model.fit_generator(
             generator=train_generator,
-            steps_per_epoch=n_samples // batch_size,
-            epochs=epochs,
+            steps_per_epoch=n_samples // batch_size, #TODO: handle case when steps_per_epoch==0
+            epochs=epochs
         )
 
     def persist(self, **kwargs):

access_niu/sample/sample_template.yml → access_niu/sample/colors/sample_template.yml

@@ -1,9 +1,9 @@
 data:
-  train: '/home/majeed/data/gender/train'
-  test: '/home/majeed/data/gender/test'
+  train: 'access_niu/sample/colors/train'
+  test: '/home/majeed/data/colors/train'
 project:
-  path: output
-batch_size: 24
+  path: 'colors'
+batch_size: 1
 epochs: 2
 pipeline:
   - name: 'mobilenet_v2'