What is the meaning of "self.max_objs = 128"

zzzxxxttt / pytorch_simple_CenterNet_45

A simple pytorch implementation of CenterNet (Objects as Points)

308 stars 63 forks source link

What is the meaning of "self.max_objs = 128" #10

Closed songyuc closed 4 years ago

songyuc commented 4 years ago

Hi guys, I am trying training this repo on my custom dataset, and I want to know what the meaning of self.max_objs = 128 is, is this mean that there must be no more than 128 objects in every image?

Your answer or idea will be appreciated!

zzzxxxttt commented 4 years ago

Yes, there are less than 128 objects in each image, at least for COCO dataset. Change this parameter to a reasonable value if there are more objects in your dataset.

songyuc commented 4 years ago

Hi, @zzzxxxttt , thanks very much for your explanation and guide!