centernet-pytorch
centernet-pytorch copied to clipboard
关于summery.py中测试CenterNet_HourglassNet
测试HourglassNet时出现了这个错误,该怎么解决呢?导儿
需要设置一下这个{'hm': self.num_classes, 'wh': 2, 'reg':2}
谢谢导儿