pvt_small is not in the backbone registry

whai362 / PVT

Official implementation of PVT series

Apache License 2.0

1.73k stars 246 forks source link

pvt_small is not in the backbone registry #36

Closed mathshangw closed 3 years ago

mathshangw commented 3 years ago

i got this error raise type(e)(f'{obj_cls.__name__}: {e}') KeyError: "RetinaNet: 'pvt_small is not in the backbone registry'" when I used python train.py configs/retinanet_pvt_s_fpn_1x_coco_640.py

whai362 commented 3 years ago

Hi, do you import pvt in your train.py? We have registered pvt_small at line #312 in https://github.com/whai362/PVT/blob/main/detection/pvt.py.