YOLOv3 with quadrangle

Reimplementation of YOLOv3 with quadrangle

This is a reimplementation of YOLOv3: An Incremental Improvement and is based on Ultralytics LLC's PyTorch implementation. This work detects obejcts in arbitrary directions with quadrangle, and implemented on ICDAR2015 text dataset for example.

Requirements

• Python3

• numpy

• torch

• opencv-python

• Shapely

Train

Check the -data_config_path and -cfg in train.py.

Dataset folder is organized as follows:

• Dataset

• images

• labels

The label format: class x1 y1 x2 y2 x3 y3 x4 y4

For example (0 493 115 519 115 519 131 493 131)

$ python3 train.py

Train for your own dataset

• Modify yolov3.cfg file

• Change [yolo] classes with the number of classes in your own dataset.

• Replace the value of filters in [convolutional] which lays above [yolo], filters should be 3 * (8 + 1 + num_classes), where 8 means 8 offsets of the quadrangle, 1 means objectness confidence.

• Modify cfg/*.data where classes field should be your number of classes in your dataset

• Modify data/*names and put your labels in it.

Inference

Checkpoints are saved in weights.

$ python3 detect.py