Fix format expected by torchvision utils

This commit is contained in:
Adam J. Stewart 2021-05-12 21:04:28 -05:00
Родитель a25e0e2238
Коммит 0f52fce0cf
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: C66C0675661156FC
2 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -2,6 +2,7 @@ black
flake8
flake8-import-order
mypy
PIL
pycocotools
rarfile
torchvision

Просмотреть файл

@ -83,7 +83,9 @@ class VHR10(VisionDataset):
"""
id = self.ids[index]
image = self._load_image(id)
target = self._load_target(id)
annot = self._load_target(id)
target = dict(image_id=id, annotations=annot)
if self.transforms is not None:
image, target = self.transforms(image, target)