YOLO Is The State-Of-The-Art, Real Time System Built On Deep Learning For Solving Object Detection Problems
YOLO Is The State-Of-The-Art, Real Time System Built On Deep Learning For Solving Object Detection Problems
@ml.india
YOLO Architecture:
@ml.india
@ml.india
pipeline to perform this task in multiple steps. This can
be slow to run and also hard to optimize, because each
individual component must be trained separately. YOLO,
does it all with a single neural network.
@ml.india
First, pretrain the first 20 convolutional layers using the
ImageNet 1000-class competition dataset, using a input
size of 224x224.
Then, increase the input resolution to 448x448.
Train the full network for about 135 epochs using a batch
size of 64, momentum of 0.9 and decay of 0.0005.
Learning rate schedule: for the first epochs, the learning
rate was slowly raised from 0.001 to 0.01. Train for about
75 epochs and then start decreasing it.
Use data augmentation with random scaling and
translations, and randomly adjusting exposure and
saturation.
@ml.india
compute the loss for the true positive, we only want one of
them to be responsible for the object. For this purpose, we
select the one with the highest IoU (intersection over union)
with the ground truth.
Classification loss.
@ml.india
@ml.india
https://medium.com/@jonathan_hui/real-time-
object-detection-with-yolo-yolov2-28b1b93e2088
https://arxiv.org/pdf/1506.02640.pdf