Проблема с изменением размера ограничивающей рамки PyTorch в формате COCO: индексы списка должны быть целыми числами или ⇐ Python

[Anonymous]

Я пробую этот пример
https://lightning.ai/lightning-ai/studi ... ng-example< /p>
И это выходной код ограничивающей рамки в аннотации изображения размером 512X512

Код: Выделить всё

bbox = dataset.coco.loadAnns(dataset.ids[2])[0]['bbox']
bbox

Вывод

Код: Выделить всё

[56.172661870503596,
120.63309352517986,
10.129496402877699,
13.812949640287755]

Я выполняю следующие строки, чтобы убедиться, что BB также обрабатывается в процессоре изображений, используя этот код

Код: Выделить всё

small = (bbox[:, 2] * bbox[:, 3]) 

Подробнее здесь: [url]https://stackoverflow.com/questions/79145935/pytorch-bounding-box-in-coco-format-resize-issue-list-indices-must-be-integers[/url]