vacancy / PreciseRoIPooling

Precise RoI Pooling with coordinate gradient support, proposed in the paper "Acquisition of Localization Confidence for Accurate Object Detection" (https://arxiv.org/abs/1807.11590).
MIT License
772 stars 152 forks source link

complie error #13

Closed BruceLee0718 closed 5 years ago

BruceLee0718 commented 5 years ago

when I run the ./ travis.sh, I got the error:

Working directory: /media/t1/0a33a0a0-b876-45c1-bb1c-f0d6fecc3bf9/cascade/bin/cascade-rcnn_Pytorch/lib/model/pr_roi_pooling/pytorch/prroi_pool/src Compiling prroi_pooling kernels by nvcc... Working directory: /media/t1/0a33a0a0-b876-45c1-bb1c-f0d6fecc3bf9/cascade/bin/cascade-rcnn_Pytorch/lib/model/pr_roi_pooling/pytorch/prroi_pool Building python libraries... Traceback (most recent call last): File "build.py", line 13, in import torch ImportError: No module named 'torch' But I think pytorch is installed correctly.Could you help me? @vacancy Thank you~

vacancy commented 5 years ago

Hi, @lichuang0529 What's the version of Python you are using? In travis.sh, I assumed that you are using python3. Please check the script for details.

BruceLee0718 commented 5 years ago

Hi, @lichuang0529 What's the version of Python you are using? In travis.sh, I assumed that you are using python3. Please check the script for details.

Hi, @lichuang0529 What's the version of Python you are using? In travis.sh, I assumed that you are using python3. Please check the script for details.

Yes ,you are right.I used the Python2.7. Thank you very much @vacancy