YOLOV3 icon indicating copy to clipboard operation
YOLOV3 copied to clipboard

Anchors setting when training on the custom dataset

Open ShaofengZou opened this issue 4 years ago • 1 comments

Hey, I find your code is clean and elegant. When training on my dataset, I have two questions about anchors:

  1. What is the relationship between these float anchors and interger anchors in comment? 1.25 is 12 scaled by 8 times? 1.875 is 36 scales by 16 times?
MODEL = {"ANCHORS":[[(1.25, 1.625), (2.0, 3.75), (4.125, 2.875)], # Anchors for small obj(12,16),(19,36),(40,28)
[(1.875, 3.8125), (3.875, 2.8125), (3.6875, 7.4375)], # Anchors for medium obj(36,75),(76,55),(72,146)
[(3.625, 2.8125), (4.875, 6.1875), (11.65625, 10.1875)]], # Anchors for big obj(142,110),(192,243),(459,401)
"STRIDES":[8, 16, 32],
"ANCHORS_PER_SCLAE":3
}
  1. Is that necessary to use kmeans algorithm to generate new anchors when training on the custom dataset?

Looking forward to your reply.

ShaofengZou avatar Jan 10 '21 01:01 ShaofengZou

你好,请问你跑自己的数据集能够跑通吗

mx985211 avatar Aug 09 '21 12:08 mx985211