Error in evaluate.py - Githubissues

vikrant7 / mobile-vod-bottleneck-lstm

Implementation of Mobile Video Object Detection with Temporally-Aware Feature Maps using PyTorch

131 stars 37 forks source link

Error in evaluate.py #18

Open AmitYadu opened 4 years ago

AmitYadu commented 4 years ago

Traceback (most recent call last): File "evaluate.py", line 227, in true_case_stat[class_index], KeyError: 1

Mindbooom commented 4 years ago

Maybe you havn't set background as one class. The classes are ['background', # always index 0 'airplane', 'antelope', 'bear', 'bicycle', 'bird', 'bus', 'car', 'cattle', 'dog', 'domestic_cat', 'elephant', 'fox', 'giant_panda', 'hamster', 'horse', 'lion', 'lizard', 'monkey', 'motorcycle', 'rabbit', 'red_panda', 'sheep', 'snake', 'squirrel', 'tiger', 'train', 'turtle', 'watercraft', 'whale', 'zebra']