GitHub - mrd0n/yoshi_detector: My implementation of a yolov5 model trained to detect my cat at the front door

mrd0n / yoshi_detector Public

Notifications You must be signed in to change notification settings
Fork 1
Star 0

My implementation of a yolov5 model trained to detect my cat at the front door

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
coco-manager @ 7948e03		coco-manager @ 7948e03
yolov5 @ c9bda11		yolov5 @ c9bda11
.gitignore		.gitignore
.gitmodules		.gitmodules
README		README
best-yoshi.pt		best-yoshi.pt
cat.jpg		cat.jpg
detect-yoshi.config.sample		detect-yoshi.config.sample
detect-yoshi.py		detect-yoshi.py
detect-yoshi.sh		detect-yoshi.sh
dons_notes.txt		dons_notes.txt
nocat.jpg		nocat.jpg

Repository files navigation

My config of YoloV5 trained to find Yoshi at the front door

Python environment:
conda activate yolov5

labelImg was used to label the training images

To Train:
python train.py --img 640 --batch 1 --epochs 200 --data ./data/yoshi.yaml --cfg ./models/yolov5m.yaml --weights '' --device 0

To Run:
start tmux
./detect-yoshi.sh