This commit is contained in:
Adam J. Stewart 2021-05-11 22:07:39 -05:00
Родитель 08608b56fb
Коммит 63b628db73
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: C66C0675661156FC
1 изменённых файлов: 9 добавлений и 1 удалений

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

@ -102,7 +102,15 @@ class VHR10(VisionDataset):
the image
"""
path = self.coco.loadImgs(id)[0]["file_name"]
return Image.open(os.path.join(self.root, path)).convert("RGB")
return Image.open(
os.path.join(
self.root,
self.base_folder,
"NWPU VHR-10 dataset",
"positive image set",
path,
)
).convert("RGB")
def _load_target(self, id: int) -> Any:
"""Load the annotations for a single image.