Open suzyahyah opened 3 years ago
Hi, I was able to run the feature extraction script, scripts/ extract_imgfeat.sh with the Docker image from https://hub.docker.com/r/chenrocks/butd-caffe/tags?page=1&ordering=last_updated
scripts/ extract_imgfeat.sh
It processed ~750 images before I encountered this:
/src/tools/../lib/rpn/proposal_layer.py:184: RuntimeWarning: invalid value encountered in greater_equal keep = np.where((ws >= min_size) & (hs >= min_size))[0] F0607 05:45:37.953456 123 roi_pooling_layer.cu:87] Check failed: error == cudaSuccess (9 vs. 0) invalid configuration argument
/src/tools/../lib/rpn/proposal_layer.py:184: RuntimeWarning: invalid value encountered in greater_equal
keep = np.where((ws >= min_size) & (hs >= min_size))[0]
F0607 05:45:37.953456 123 roi_pooling_layer.cu:87] Check failed: error == cudaSuccess (9 vs. 0) invalid configuration argument
Do you have any pointers on what I could possibly try to fix this? Thank you!
Hi, I was able to run the feature extraction script,
scripts/ extract_imgfeat.shwith the Docker image from https://hub.docker.com/r/chenrocks/butd-caffe/tags?page=1&ordering=last_updatedIt processed ~750 images before I encountered this:
/src/tools/../lib/rpn/proposal_layer.py:184: RuntimeWarning: invalid value encountered in greater_equalkeep = np.where((ws >= min_size) & (hs >= min_size))[0]F0607 05:45:37.953456 123 roi_pooling_layer.cu:87] Check failed: error == cudaSuccess (9 vs. 0) invalid configuration argumentDo you have any pointers on what I could possibly try to fix this? Thank you!