/tensorflow-yolo3

tensorflow implementation of yolov3

Primary LanguagePython

tensorflow-yolo3

license


Detection

1、If use the pretrain model, download YOLOV3 weights from YOLO website.
2、Modify yolo3_weights_path in the config.py
3、Run detect.py

wget https://pjreddie.com/media/files/yolov3.weights  
python detect.py --image_file ./test.jpg  

result

Training

convert train and val data to tfrecord

1、Download the COCO2017 dataset from COCO_website
2、Modify the train and val data path in the config.py
3、If you want to use original pretrained weights for YOLOv3, download from darknet53 weights
4、rename it as darknet53.weights, and modify the darknet53_weights_path in the config.py

wget https://pjreddie.com/media/files/darknet53.conv.74`  

4、Modify the data augmentation parameters and train parameters
5、Run yolo_train.py

Evaluation

1、Modify the pre_train_yolo3 and model_dir in config.py
2、Run detect.py

python detect.py --image_file ./test.jpg

Train Image show on Tensorboard

train

Notice

If you want to modify the Gpu index, please modify gpu_index in config.py

Credit

@article{yolov3,
	title={YOLOv3: An Incremental Improvement},
	author={Redmon, Joseph and Farhadi, Ali},
	journal = {arXiv},
	year={2018}
}

Reference