hanoonaR/object-centric-ovd

Error during class-specific IN-21K proposal generation for LVIS

Closed this issue · 0 comments

Hi,
While generating class-specific ILS proposals for ImageNet-21k using get_ils_labels.py, one might encounter the following error:

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

This error occurs for some ImageNet-21k images which are not RGB by default.

To fix this, replace this line of modulated_detection.py (directory path: object-centric-ovd/external/mavl/inference)
with im = Image.open(image_path).convert('RGB'). It will make sure every loaded image is converted to RGB format.