def __getitem__(self, i: int) -> Tuple[torch.Tensor, torch.Tensor]:
_id = self.__image_ids[i]
paths = self.__id2path(_id)
print("step -1")
image = self.__load_n_preprocess(paths["image_path"])
mask = self.__load_n_preprocess(paths["mask_path"])
print("step 0")
if self.__transforms is not None:
segmap = SegmentationMapsOnImage(mask, shape=mask.shape)
print("step 1")
image, mask = self.__transforms(image=image, segmentation_maps=segmap)
print("step 2")
image = torch.from_numpy(image)
mask = torch.from_numpy(mask)
image = image.permute(2, 0, 1)
mask = mask.permute(2, 0, 1)
print("step 3")
return image, mask
ERROR
ValueError: Got dtype 'float64', which is a forbidden dtype (bool, uint16, uint32, uint64, uint128, uint256, int32, int64, int128, int256, float16, float32, float64, float96, float128, float256).
‘data.py’ file includes function above. I detected error with my extraordinary debug style I cannot see “step 2” in results. What should I do ?
image, mask = self.__transforms(image=image, segmentation_maps=segmap)
Thanks in advance!