def __getitem__(self, item):
row, col, label = self.coordinate_label[:, item]
# row, col = self.indexToCoordinate(row_col, self.cube.shape[0])
spectral_cube_s = getCubeByCoordinate(self.cube, int(row), int(col),
self.cube_radius)
spectral_cube_s = self.totensor(spectral_cube_s)
return spectral_cube_s, int(label)
def __getitem__(self, item):
row, col, label = self.coordinate_label[item]
spectral_cube_s = getCubeByCoordinate(self.cube, int(row), int(col),
self.cube_radius)
spectral_cube_s = self.totensor(spectral_cube_s)
if self.test:
return spectral_cube_s, int(label), row, col
return spectral_cube_s, 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)
def __getitem__(self, item):
row, col, label = self.coordinate_label[item]
spectral_cube = np.transpose(
getCubeByCoordinate(self.cube, int(row), int(col), radius=16),
(2, 0, 1))
if self.transform != None:
spectral_cube = self.transform(spectral_cube)
if self.test:
return spectral_cube, int(label), row, col
return spectral_cube, int(label)
def __getitem__(self, item):
row, col, label = self.coordinate_label[item]
spectral_cube_s = getCubeByCoordinate(self.cube, int(row), int(col),
self.cube_radius)
spectral_cube_s = torch.as_tensor(spectral_cube_s,
dtype=torch.double).reshape(
-1, spectral_cube_s.shape[-1])
spectral_cube_s = torch.mm(self.neigh_weight, spectral_cube_s)
if self.test:
return spectral_cube_s, int(label), row, col
return spectral_cube_s, 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]
spectral_cube_s = getCubeByCoordinate(self.cube, row, col,
self.cube_radius - 2)
spectral_cube_s = self.totensor(spectral_cube_s)
return spectral_cube_s, int(label)
