YOLOSO

A Modified YOLOv2 for Sharp Object Detection

models

all models can be found in model_ref.py

tensorboard

tensorboard --logdir=tensorboard/yolo_voc/sharp_object_dataset

training

python3 train.py --num_epoches 1000 --pre_trained_model_type model0

evaluation

python test_dataset.py python eval_voc.py

Acknowledgement

This project is based on the following projects. Thank you for the wonderful work.

yolo bootstrapped from:
https://github.com/uvipen/Yolo-v2-pytorch

mAP boostrapped from:
https://github.com/argusswift/YOLOv4-pytorch/

labelling:
https://github.com/tzutalin/labelImg

feature maps:
https://ravivaishnav20.medium.com/visualizing-feature-maps-using-pytorch-12a48cd1e573