facebookresearch / facebookresearch/detectron2

visualize_json_results.py IndexError: too many indices for tensor of dimension 1

Open
#2,763 2 comments 0 reactions 0 assignees View on GitHub
contributions welcome enhancement
Dominant language
Python
Stars
34.7k
Forks
7.9k
PR merge metrics
No merged PRs in 30d

Description

## Instructions To Reproduce the 🐛 Bug:
I use visualize_json_results.py to visualize the results for faster rcnn on lvis_v0.5_val dataset, however I can only get results for 12 pictures and get the following error:
![image](https://user-images.githubusercontent.com/39671748/111638637-7c00bf00-8835-11eb-8325-0144c1122c39.png)

## Expected behavior:

I want to get the visualization results for every picture in lvis_v0.5_val dataset.

## Environment:
detectron2_v0.1.1

![image](https://user-images.githubusercontent.com/39671748/111637860-c59cda00-8834-11eb-87d2-42c600dd1534.png)

Contributor guide

Open the contributing guide

Research direction

Start with visualize_json_results.py and reproduce the failure using Faster R-CNN results on the lvis_v0.5_val dataset under detectron2 v0.1.1. Inspect the traceback at the tensor indexing operation after the first 12 images, then verify the script can produce visualization results for every picture in the dataset.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
computer-vision
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.