/scaledyolov4-cpu

A cpu supported variant of Scaled YOLOv4 CSP

Primary LanguagePython

YOLOv4-CSP

This is a cpu only fork of Scaled-YOLOv4-csp, published as "Scaled-YOLOv4: Scaling Cross Stage Partial Network".

Changes

Replaced Cuda-Mish with CPU supported Mish activation function.

Run

Download yolov4-csp.weights file and save it in weights/ folder.

In a terminal run:

python inference.py --cfg models/yolov4-csp.cfg --weights weights/yolov4-csp.weights --img-src <path/to/an/image>

Reference

@article{wang2020scaled,
  title={{Scaled-YOLOv4}: Scaling Cross Stage Partial Network},
  author={Wang, Chien-Yao and Bochkovskiy, Alexey and Liao, Hong-Yuan Mark},
  journal={arXiv preprint arXiv:2011.08036},
  year={2020}
}