Skip to content

Instantly share code, notes, and snippets.

@YimianDai
Created July 30, 2019 00:22
Show Gist options
  • Save YimianDai/e8fc2df5ab477c208d8ece64a5425cfd to your computer and use it in GitHub Desktop.
Save YimianDai/e8fc2df5ab477c208d8ece64a5425cfd to your computer and use it in GitHub Desktop.
YOLO

YOLO v1

YOLO 太典型的 one-stage method,可谓简单粗暴

放弃了 Region Proposal,直接划分为 S * S 个grid

最后从 S * S * 1024 的 Feature Map 是由两个全连接层来实现的,如果要问凭什么最后 S * S * (B * 5 + 2) 的 不同的位置的映射结果不一样,那只能说是因为 label 里面给的不一样,会去学类似 label 的分布,所以在 YOLO v1 中是没有像后面两个版本因为采用 conv 来做 predition 有最终的 feature map 上的位置 和 最后的 predition 位置相对应的问题。

YOLO v2

  1. Batch Normalization.
  2. Convolutional With Anchor Boxes. 之前是 bbox,这回事 anchor 感觉没啥差别啊,YOLOv2 移除了 YOLOv1 中的全连接层而采用了卷积和 anchor boxes 来预测边界框,YOLO v1 是用全连接来预测
  3. Dimension Clusters 决定 anchor box 尺寸
  4. Direct location prediction,之前的方法都是预测 offset,但是 YOLO 还是 predict location coordinates relative to the location of the grid cell
  5. Fine-Grained Features,引入 passthrough layer,类似于 resnet,
  6. Multi-Scale Training.
  7. 用 Darknet-19 作为 backbone,更快的特征提取

YOLO v3

不用担心 YOLO v3 只标记了一个 anchor 作为 Positive,作者在论文里说了 If a bounding box prior is not assigned to a ground truth object it incurs no loss for coordinate or class predictions, only objectness. 因为计算 loss 的时候,只计算 label 是 Positive 的 location loss 和 Classification loss

  1. YOLOv3 predicts an objectness score for each bounding box using logistic regression
  2. Class Prediction 用的是 independent logistic classifiers
  3. Predictions Across Scales: YOLOv3 predicts boxes at 3 different scales, as SSD does; 而且用的是 FPN 的方式,We also take a feature map from earlier in the network and merge it with our upsampled features using element-wise addition.
  4. Change Feature Extractor from Darknet-19 to Darknet-53

问题

  1. 唯一的 Positive Anchor 不影响 bbox 的 loss,但为什么 会不影响 Classification loss 呢?
  2. Anchor box 的中心该是 Iceberg 还是 cell,应该是 cell
  3. 考虑一个 cell 里面有多个 iceberg 的问题,这个时候 anchor box 怎么 assign,这个是 iceberg 和 VOC 很大的一个差别
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment