Skip to content

qwerasdf887/TF2_TinyYOLOv4

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TF2_TinyYOLOv4

TinyYOLOv4 Tensorflow2-keras version
Tiny YOLOv4的tf.keras版本,如果需要加入DropBlock可以參考TF2_YOLOv4
Tiny v4,使用與v3相同的方式:
Conv2D -> BN -> Leakly,當作一個Conv單元

環境 environment

  1. Tensorflow 2.2
  2. tensorflow_addons (moving average opt)
  3. Python 3.5~3.7
  4. OpenCV 3~4

Tiny Weights

ALexeyAB : download weight

Predict Img:

python predict.py

Result

Training

需修改train.py,沒有寫得很完整,有問題可以發issue。 使用labelImg 生成的xml檔,並且放入標籤位置即可。

About

TinyYOLOv4 Tensorflow2-keras version

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages