YOLOv3端到端训练

YOLOv3端到端训练

paddle API fluid.layers.yolov3_loss中的参数含义:x表示输出特征图的一个层级(p0 p1 p2中的一个);gt_box表示真实框; gt_label表示真实标签 gt_score 表示一种训练技巧(有一个加mixup的训练技巧),可以设置为1,形状同gt_label;anchors表示9钟anchor的大小; anchor_mask 是会选对这一层级的futuremap需要哪种尺寸的anchor;class_num表示分类数目(注意带不带背景,在coco数据集上不带背景设置成80); ignore_thresh表示iou大于哪个阈值就把它标成-1的那个阈值,一般设置成0.7; downsample_ratio表示特征图步幅,例如p0设置成是32; use_label_smooth一种训练技巧,可以设置成False。(这个API代码居然是c++写的。。。)

YOLOv3端到端训练

上图中的anchor_masks=[[6,7,8],[],[]] 中的6 7 8 分别表示选择哪种anchor,分别选择了第6个 第7个 第8个anchor(对应anchors中画红圈的参数,因为anchor两个参数表示一个anchor大小)。

YOLOv3端到端训练