Open xll426 opened 4 years ago
File "predict.py", line 17, in r_image = efficientdet.detect_image(image) File "/home/404/efficientdet-pytorch-master/efficientdet.py", line 109, in detect_image detection = torch.cat([regression,classification],axis=-1) TypeError: cat() got an unexpected keyword argument 'axis' (yp) [root@localhost efficientdet-pytorch-master]#
版本,torch1.2.0
File "predict.py", line 17, in
r_image = efficientdet.detect_image(image)
File "/home/404/efficientdet-pytorch-master/efficientdet.py", line 109, in detect_image
detection = torch.cat([regression,classification],axis=-1)
TypeError: cat() got an unexpected keyword argument 'axis'
(yp) [root@localhost efficientdet-pytorch-master]#