TinyYOLOv4 Tensorflow2-keras version
Tiny YOLOv4的tf.keras版本,如果需要加入DropBlock可以參考TF2_YOLOv4
Tiny v4,使用與v3相同的方式:
Conv2D -> BN -> Leakly,當作一個Conv單元
- Tensorflow 2.2
- tensorflow_addons (moving average opt)
- Python 3.5~3.7
- OpenCV 3~4
ALexeyAB : download weight
python predict.py
需修改train.py,沒有寫得很完整,有問題可以發issue。 使用labelImg 生成的xml檔,並且放入標籤位置即可。