return {
'target_single_stroke': target_single_image,
'reference_single_stroke': reference_single_image,
'target_data': target_image,
'reference_color': reference_color_image,
'stroke_num': stroke_num,
'reference_single_stroke_centroid': reference_single_centroid,
# 'seg_id': stroke_label,
'stroke_label': stroke_label,
Traceback (most recent call last):
File "main_train.py", line 34, in
train_.train()
File "main_train.py", line 20, in train
self.train_sdnet.calculate_prior_information_and_qualitative(save_path='dataset_forSegNet_ExtractNet_'+self.dataset)
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/train_SDNet.py", line 360, in calculate_prior_information_and_qualitative
transformed_reference_color, transformed_single_reference_stroke = self._calculate_linear_transformation_inference(
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/train_SDNet.py", line 140, in calculate_linear_transformation_inference
linear_grid = self.sd_net.get_linear_estimation(reference_single_stroke[0], grid[0],
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/model/model_of_SDNet.py", line 80, in get_linear_estimation
grid = torch.transpose(grids, 2, 3)
IndexError: Dimension out of range (expected to be in range of [-3, 2], but got 3)
return {
'target_single_stroke': target_single_image,
'reference_single_stroke': reference_single_image,
'target_data': target_image,
'reference_color': reference_color_image,
'stroke_num': stroke_num,
'reference_single_stroke_centroid': reference_single_centroid,
# 'seg_id': stroke_label,
'stroke_label': stroke_label,
Traceback (most recent call last):
File "main_train.py", line 34, in
train_.train()
File "main_train.py", line 20, in train
self.train_sdnet.calculate_prior_information_and_qualitative(save_path='dataset_forSegNet_ExtractNet_'+self.dataset)
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/train_SDNet.py", line 360, in calculate_prior_information_and_qualitative
transformed_reference_color, transformed_single_reference_stroke = self._calculate_linear_transformation_inference(
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/train_SDNet.py", line 140, in calculate_linear_transformation_inference
linear_grid = self.sd_net.get_linear_estimation(reference_single_stroke[0], grid[0],
File "/mnt/big_disk/gbw/projects/StrokeExtraction-master_copy/model/model_of_SDNet.py", line 80, in get_linear_estimation
grid = torch.transpose(grids, 2, 3)
IndexError: Dimension out of range (expected to be in range of [-3, 2], but got 3)