RuntimeError: The shape of the mask [1, 1, 128, 128] at index 1 does not match the shape of the indexed tensor [1, 80, 128, 128] at index 1 about centernet

[deprecatedwarrior]

In models/CenterNet-104.py, model class definition, the out_dim is hard-coded to 80. For custom datasets with a different out_dim, these need to be changed along with the above-mentioned changes. Just making a note for other users.

[li10141110]

I replaced the content of my .data/coco/annotations/instances_{minival2014,trainval2014}.json to my coco style file and replaced the content of my .data/coco/images/{minival2014,trainval2014} to my own images.
following is my CenterNet-104.json file.
{
"system": {
"dataset": "MSCOCO",
"batch_size": 1,
"sampling_function": "kp_detection",

    "train_split": "trainval",
    "val_split": "minival",

    "learning_rate": 0.00025,
    "decay_rate": 10,

    "val_iter": 200,

    "opt_algo": "adam",
    "prefetch_size": 6,

    "max_iter": 100000,
    "stepsize": 4500,
    "snapshot": 5000,

    "chunk_sizes": [1],

    "data_dir": "../data"
},

"db": {
    "rand_scale_min": 0.6,
    "rand_scale_max": 1.4,
    "rand_scale_step": 0.1,
    "rand_scales": null,

    "rand_crop": true,
    "rand_color": true,

    "border": 128,
    "gaussian_bump": true,

    "input_size": [511, 511],
    "output_sizes": [[128, 128]],

    "test_scales": [1],

    "top_k": 70,
    "categories": 1,
    "kp_categories": 1,
    "ae_threshold": 0.5,
    "nms_threshold": 0.5,

    "max_per_image": 100
}

}

[Duankaiwen]

@li10141110 How many categories do you have？

[Duankaiwen]

@li10141110 Add these codes above the
dist = tag_mean.unsqueeze(1) - tag_mean.unsqueeze(2)
in models/py_utils/kp_utils.py:

if len(tag_mean.size()) < 2:

      tag_mean = tag_mean.unsqueeze(0)`  

[li10141110]

1 category

[li10141110]

Thank you Kaiwen, I make it

[nassarofficial]

Did it work with you? I had the same problem but the solution was to edit the categories number in the config file. I am also training with 1 class but I am having problems in the evaluation.

I have only 1 class, and it's giving me a "key error 0"

        category       = self._coco_to_class_map[cat_id]

in the db/coco.py file