Model | loss | accuracy |
---|---|---|
yolov8 | 1.118 | 0.7219 |
DTRB | 3.998 | 73.918 |
Run this command:
pip install -r requirements.txt
Download dataset Download pre-trained weights
Run this command:
python main.py --input-image YOUR_IMAGE.jpg
Run this command:
python verification.py --input-image YOUR_IMAGE.jpg
Also you can chang threshold using--threshold