Peterisfar / YOLOV3

yolov3 by pytorch
MIT License
195 stars 53 forks source link

Anchors setting when training on the custom dataset #43

Open ShaofengZou opened 3 years ago

ShaofengZou commented 3 years ago

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
    }
  2. Is that necessary to use kmeans algorithm to generate new anchors when training on the custom dataset?

Looking forward to your reply.

mx985211 commented 3 years ago

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