Merge pull request #35 from saaresearch/develop

Update Readme and fit bug image in example folder

README.md

@@ -173,6 +173,7 @@ After you have placed your dataset in the folder ***user_datasets/*** and create
 
     EPOCHS: 2
     IMG_SIZE: 300
+    BATCH_SIZE: 16
 
     # MODEL ZOO:
     # ["yolov5l", "yolov5m", "yolov5n", "yolov5s", "yolov5x",
@@ -182,8 +183,10 @@ After you have placed your dataset in the folder ***user_datasets/*** and create
     # **NOTE**: For successful training of the ssd model, 
     # the size of your images should not exceed 512x512
 
-    MODEL: ssd
 
+    # You can also use this format if need to train few models
+    # MODEL: ['yolov5s', 'yolov5l'] this can also be applied to parameters EPOCHS, BATCH_SIZE, IMG_SIZE
+    MODEL: ssd
 
     # For multiprocessing.
     # For CPU:

examples/api_example_ODRS.ipynb

@@ -246,7 +246,7 @@
         "id": "QyNYYnRb-yP8"
       },
       "source": [
-        "![WaRP-Categories.png](attachment:WaRP-Categories.png)"
+        "![WaRP-Categories.png](../docs/img/WaRP-Categories.png)"
       ]
     },
     {
@@ -264,7 +264,7 @@
         "id": "CmBW9iRq-yP8"
       },
       "source": [
-        "![WaRP-Dataset.png](attachment:WaRP-Dataset.png)"
+        "![WaRP-Dataset](../docs/img/WaRP-Dataset.png)"
       ]
     },
     {