-
Download the dataset here
-
and then set the path to be like this
./
|
├─── dataset/
| ├─── test/
| ├─── train/
| └─── ...
└───...- Change the
OUTPUT_DIRvariable in the ipynb file
OUTPUT_DIR = "./model"- Download the dependencies
uv sync-
Run the
.ipynbfile to get the model weight -
Make sure you already have the models in
modelfolder
./
|
├─── model/
| ├─── best_resnet_yolo.pth
| ├─── label_encoder.pkl
| └─── yolo12n.pt
└───...- Run the
test-app.py
uv run test-app.py- The result should be looked like this
