def __getitem__(self, item):
row, col, label = self.coordinate_label[:, item]
spectral_vector = getVectorByCoordinate(self.cube, int(row), int(col))
if self.transform != None:
spectral_vector = self.transform(spectral_vector)
return spectral_vector, int(label)
def __getitem__(self, item):
row, col, label = self.coordinate_label[item]
spectral_vector = getVectorByCoordinate(self.cube, int(row), int(col))
pca_cube = getCubeByCoordinate(self.pca_cube,
int(row),
int(col),
radius=30)
if self.transform != None:
spectral_vector = self.transform(spectral_vector)
if self.test:
return spectral_vector, pca_cube, int(label), row, col
return spectral_vector, pca_cube, int(label)
def __getitem__(self, item):
row, col, label = self.coordinate_label[item]
# row, col = self.indexToCoordinate(row_col, self.cube.shape[0])
# gt = self.gt[row, col]
# if int(gt) == int(label):
# print("yes")
# else:
# print("no ", "gt: ", gt, " label: ", label)
spectral_vector = getVectorByCoordinate(self.cube, int(row), int(col))
spectral_cube_b = getCubeByCoordinate(self.cube, row, col,
self.cube_radius)
spectral_cube_s = getCubeByCoordinate(self.cube, row, col,
self.cube_radius - 2)
if self.transform != None:
spectral_vector = self.transform(spectral_vector)
spectral_cube_b = self.totensor(spectral_cube_b)
spectral_cube_s = self.totensor(spectral_cube_s)
return spectral_vector, spectral_cube_b, spectral_cube_s, int(label)