Computer-Vision-Team-Amsterdam / Detecting-Heavy-Objects

0 stars 0 forks source link

Fix no segementation mask in postprocessing.py #89

Closed chrise96 closed 1 year ago

chrise96 commented 1 year ago

Thomas found this bug when reading the following data in postprocessing.py:

[{"pano_id": "TMX7316010203-001697_pano_0000_000217.jpg", "category_id": 0, "bbox": [0.0, 0.0, 8000.0, 4000.0], "score": 1.0}, {"pano_id": "TMX7316010203-001697_pano_0000_000216.jpg", "category_id": 0, "bbox": [0.0, 0.0, 8000.0, 4000.0], "score": 1.0}, {"pano_id": "TMX7316010203-001697_pano_0000_000215.jpg", "category_id": 0, "bbox": [0.0, 0.0, 8000.0, 4000.0], "score": 1.0}, {"pano_id": "TMX7316010203-001697_pano_0000_000220.jpg", "category_id": 0, "bbox": [0.0, 0.0, 8000.0, 4000.0], "score": 1.0}]

There is no segmentation key in this file, the following code will fail: https://github.com/Computer-Vision-Team-Amsterdam/Detecting-Heavy-Objects/blob/development/postprocessing.py#L192