No module named cython_bbox

Hi, I’m trying your code but when I run:

CUDA_VISIBLE_DEVICES=$GPU_ID python trainval_net.py --dataset pascal_voc --net vgg16 --cuda --bs $BATCH_SIZE

for the training I have this error:

from model.utils.cython_bbox import bbox_overlaps ImportError: No module named cython_bbox

I’ve just installed cython doing: sudo pip install cython

Thanks in advance

[pranerd]

did you cd lib directory and run the command sh make.sh?

[jwyang]

The minimal batch size should be 2. If you want make it work, you can reduce the image size, and use resnet instead of vgg16.

On Wed, Jan 10, 2018 at 18:35 Capuz93 [e-mail] wrote:

I changed the make file and I have a warning but I think it’s ok: nvcc warning : The ‘compute_20’, ‘sm_20’, and ‘sm_21’ architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).

Then when I run the program setting BATCH_SIZE=2 and RPN_BATCHSIZE=2 I have “out of memory” error; when I run the program setting BATCH_SIZE=1 and RPN_BATCHSIZE=2 I have this error:

File “/home/Scrivania/TESI/FASTER_RCNN/2/faster-rcnn.pytorch-master/lib/model/rpn/proposal_target_layer_cascade.py”, line 168, in _sample_rois_pytorch rand_num = torch.from_numpy(rand_num).type_as(gt_boxes).long() RuntimeError: the given numpy array has zero-sized dimensions. Zero-sized dimensions are not supported in PyTorch

— You are receiving this because you modified the open/close state.

Reply to this email directly, view it on GitHub https://github.com/jwyang/faster-rcnn.pytorch/issues/9#issuecomment-356773953, or mute the thread https://github.com/notifications/unsubscribe-auth/ADtr55iF19TvZeRQ79TvRDxr6zCHI2W0ks5tJUk-gaJpZM4ROMy0 .