efficientdet-pytorch icon indicating copy to clipboard operation
efficientdet-pytorch copied to clipboard

bug

Open xll426 opened this issue 4 years ago • 1 comments

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]#

xll426 avatar Jul 16 '20 01:07 xll426

版本,torch1.2.0

bubbliiiing avatar Jul 17 '20 05:07 bubbliiiing