search

bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码，可以用于训练自己的模型。
MIT License
305 stars 61 forks source link

bug #1

Open xll426 opened 4 years ago

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

bubbliiiing commented 4 years ago

版本，torch1.2.0