How do you change num of classes?

[sarmientoj24]

How do I change number of classes here? And say, I have 17 classes, what should be the number?

[zzzxxxttt]

just add the following lines to the config file

MODEL:
    RETINANET:
        NUM_CLASSES: 17

[sarmientoj24]

i shall be using efficientdet. this is for that too, right?

[zzzxxxttt]

yes, it's based on retinanet

[sarmientoj24]

thank you. can i know what is the architecture of the whole efficientdet?

[sarmientoj24]

hi, this doesnt support non-squared images?

[zzzxxxttt]

The data preprocessing pads a non-squared image to squared, if you need to use non-squared input, I think the height and width should be divisible by 256.