longcw / faster_rcnn_pytorch

Faster RCNN with PyTorch
MIT License
1.7k stars 466 forks source link

Segmentation fault #38

Closed bywbilly closed 6 years ago

bywbilly commented 6 years ago

Hi, I use python2.7 and pytorch 0.1.12.post2 with cuda 8.0.

Then I run the python train.py. I got something like that.

image

Here is the gdb results:

image

Tree Trace: image

Could anybody have some ideas? Seems like a opencv issue?

Thank you!

bywbilly commented 6 years ago

BTW, I find that the problematic code seems to be:

image in faster_rcnn/utils/blob.py

bywbilly commented 6 years ago

Maybe it's the bug caused by opencv2. I could successfully train and test by resizing the image with scipy.