yolo2-pytorch
yolo2-pytorch copied to clipboard
YOLOv2 in PyTorch
Traceback (most recent call last): File "demo.py", line 37, in net_utils.load_net(trained_model, net) File "/home/xzm_project/DLpj/yolo/yolo2-pytorch/utils/network.py", line 66, in load_net param = torch.from_numpy(np.asarray(h5f[k])) File "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper File "h5py/_objects.pyx", line...
Traceback (most recent call last): File "demo.py", line 37, in net_utils.load_net(trained_model, net) File "/home/pcrane/yolo2-pytorch/utils/network.py", line 66, in load_net param = torch.from_numpy(np.asarray(h5f[k])) File "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper File "h5py/_objects.pyx", line...
demo
ImportError: /home/zgx/mask_recognition/yolo2-pytorch-master/layers/reorg/_ext/reorg_layer/_reorg_layer.so: undefined symbol: __cudaRegisterFatBinaryEnd
cpu only
I want to use it with pytorch-cuponly ,what should I do? thanks!!
I need the version number as detailed as possible! like Cython==0.27.3,but how about python,pytorch ?I don't know! thanks!!!
I run in multi (2~4) GPUs (Tesla V100), there are something error. Found GPU0 V100-SXM2-16GB which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time, but your PyTorch...
Hi, I'm confused about the problem that there is no module named 'utils.nms.gpu_nms'
I tried python demo under windows7 python 3.6 torch0.4.1 but it showed the error Traceback (most recent call last): File "demo.py", line 6, in from darknet import Darknet19 File "C:\LPR\yolo2-pytorch\darknet.py",...