search

zzzxxxttt / pytorch_simple_CenterNet_45

A simple pytorch implementation of CenterNet (Objects as Points)
308 stars 64 forks source link

IndexError: index 100 is out of bounds for axis 0 with size 100 #26

Open Shank2358 opened 4 years ago

Shank2358 commented 4 years ago

line 167, in getitem wh[k] = 1. w, 1. h IndexError: index 100 is out of bounds for axis 0 with size 100

GerardWalsh commented 3 years ago

Same issue when training on a custom dataset with one class.

GerardWalsh commented 3 years ago

I was trying to use the pascal dataloader for training on a custom dataset with 1 class - have switched to coco dataset loader and have left self.max_objs = 128, and the error IndexError: index 100 is out of bounds for axis 0 with size 100 is no longer occuring.