Downloads · 30 days
0
yolov7-lego
mw00/yolov7-lego
yolov7-lego is a object detection model from mw00. Use it when you need objects located in an image. The card lists the license as cc0-1.0.
The model(s) in this repository are trained with the dreamfactor/biggest-lego-dataset-600-parts from Kaggle and the Yolov7 training script.
Access
Public
Updated Jul 28, 2023
Repo size
12.3 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt12.3 MB · 92%
At a glance
- Task
- Object Detection
- License
- cc0-1.0
- Access
- Public
- Created
- Jul 26, 2023
- Updated
- Jul 28, 2023
- SHA
- 07035cb9
Spaces
Model card
From the Hugging Face model README
Overview
The model(s) in this repository are trained with the dreamfactor/biggest-lego-dataset-600-parts from Kaggle and the Yolov7 training script.
Limitations
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).
- Task
- Object Detection
- License
- cc0-1.0
- Created
- Jul 26, 2023
- Updated
- Jul 28, 2023