ChenRocks / ChenRocks/UNITER

feature extraction with docker: RuntimeWarning: invalid value encountered in greater_equal

Open
#76 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
799
Forks
111
PR merge metrics
No merged PRs in 30d

Description

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

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`

Do you have any pointers on what I could possibly try to fix this? Thank you!

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reproducing the failure with scripts/ extract_imgfeat.sh in the referenced Docker image. Inspect proposal_layer.py:184 and roi_pooling_layer.cu:87 around the reported warning and CUDA error. Done means feature extraction processes the image set without the runtime warning or invalid-configuration crash.

Written by the indexing model from the issue text.

Assessment

Tech stack
docker, python
Domain
computer-vision, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.