Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
conan7882 authored Nov 7, 2019
1 parent 2b65f23 commit 669161f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# YOLOv3 for Object Detection

- TensorFlow implementation of [YOLOv3](https://arxiv.org/abs/1804.02767) for object detection.
- TensorFlow implementation of [YOLOv3](https://pjreddie.com/media/files/papers/YOLOv3.pdf) for object detection.
- Both [inference](https://github.com/conan7882/YOLOv3/tree/master#use-pre-trained-model-for-object-detection-80-classes) and [training](https://github.com/conan7882/YOLOv3/tree/master#train-on-voc2012-dataset-20-classes) pipelines are implemented.
- For inference using pre-trained model, the model stored in `.weights` file is first downloaded from official [YOLO website](https://pjreddie.com/darknet/yolo/) (Section 'Performance on the COCO Dataset', YOLOv3-416 [link](https://pjreddie.com/media/files/yolov3.weights)), then [converted](docs/convert.md) to `.npy` file and finally loaded by the TensorFlow model for prediction.
- For training, the pre-trained DarkNet-53 is used as the feature extractor and the YOLO prediction layers at three scales are trained from scratch. Data augmentation such as random flipping, cropping, resize, affine transformation and color change (hue, saturation, brightness) are applied. Anchor clustering and multiple scale training (rescale training images every 10 epochs) are implemented as well.
Expand Down

0 comments on commit 669161f

Please sign in to comment.