Refactoring and renaming

README.md

@@ -4,7 +4,6 @@ pipeline_tag: object-detection
 tags:
 - lego
 - brick
-- bricks
 ---
 
 # Overview
@@ -14,7 +13,7 @@ from Kaggle and the [Yolov7](https://github.com/WongKinYiu/yolov7) training scri
 
 ## Limitations
 
-The `zero-shot-1000.pt` was trained in the `lego-training-1000.ipynb` notebook with 1000 images and does not differentiate lego classes but only tries to predict Lego objects.
+The `zero-shot-1000-single-class.pt` was trained in the `training-zero-shot-1000-single-class.ipynb` notebook with 1000 images and does not differentiate lego classes but only tries to predict Lego objects.
 This can be easily reconfigured and retrained in the notebook, but the current implementation leads to many false positives on non-Lego objects and therefore can be improved
 upon. Also, it could be worth investigating if the metrics improve with a bigger training dataset, as currently only 1000 images are being used (approx. 0.6% of the full 
 dataset).