theAIGuysCode / Object-Detection-API

Yolov3 Object Detection implemented as APIs, using TensorFlow and Flask
Apache License 2.0
341 stars 180 forks source link

Very weak results with Tiny YOLOv3 weights #8

Closed Gooogr closed 4 years ago

Gooogr commented 4 years ago

Hello! I got very bad detection result for tiny YOLO model, it feels like model is highly underfitted. I used detection.py file and changed strings 14 - 16 to use tiny version like:

flags.DEFINE_string('weights', './weights/yolov3-tiny.tf',
                    'path to weights file')
flags.DEFINE_boolean('tiny', True, 'yolov3 or yolov3-tiny')

The weights of models were downloaded by links from READ.ME file and were converted to tensorflow format without any errors.

Examples with YOLOv3 and Tiny YOLO v3: dog_yolo dog_tiny_yolo

crowd_yolo crowd_tiny_yolo

smartphones_yolo smartphones_tiny_yolo

Gooogr commented 4 years ago

After some tests, I understood those results are normal for tiny YOLO architecture. You can improve it if you will use custom train anchors and play around with the detection threshold. In this case, use AlexeyAB`s realization, not pjreddie's one.