bottom-up-attention.pytorch icon indicating copy to clipboard operation
bottom-up-attention.pytorch copied to clipboard

Number of bounding boxes not consistent between extract_features.py and visualize.ipynb

Open preetom-saha-arko opened this issue 2 years ago • 0 comments

I ran the following command to extract features from an image: python3 extract_features.py --mode caffe --num-cpus 0 --gpus 1 --extract-mode roi_feats --min-max-boxes '1,70' --config-file configs/caffe/test-caffe-r101.yaml --image-dir "/home/arko/Documents/images" --out-dir "/home/arko/Documents/outputs"

(There is only one image in my image directory.)

After extracting the archive, I get multiple npy files. Reading the num_bbox.npy file, I get 63.

However, using the same model in visualize.ipynb with CONF_THRESH = 0.3. MIN_BOXES = 1 and MAX_BOXES = 70, I am getting 50 bounding boxes for the same image.

Why is there such discrepancies? Am I missing something?

preetom-saha-arko avatar Feb 01 '23 04:02 preetom-saha-arko