yuxin
commited on
Commit
·
53b47bc
1
Parent(s):
2f9c026
add model
Browse files- model_segvol_single.py +1 -2
model_segvol_single.py
CHANGED
|
@@ -76,12 +76,11 @@ class SegVolProcessor():
|
|
| 76 |
item['label'] = gt_voxel_ndarray.astype(np.int32)
|
| 77 |
|
| 78 |
# transform
|
|
|
|
| 79 |
item = self.transform(item)
|
| 80 |
item_zoom_out = self.zoom_out_transform(item)
|
| 81 |
item['zoom_out_image'] = item_zoom_out['image']
|
| 82 |
item['zoom_out_label'] = item_zoom_out['label']
|
| 83 |
-
del item['image_transforms']
|
| 84 |
-
del item['label_transforms']
|
| 85 |
return item
|
| 86 |
|
| 87 |
|
|
|
|
| 76 |
item['label'] = gt_voxel_ndarray.astype(np.int32)
|
| 77 |
|
| 78 |
# transform
|
| 79 |
+
return item
|
| 80 |
item = self.transform(item)
|
| 81 |
item_zoom_out = self.zoom_out_transform(item)
|
| 82 |
item['zoom_out_image'] = item_zoom_out['image']
|
| 83 |
item['zoom_out_label'] = item_zoom_out['label']
|
|
|
|
|
|
|
| 84 |
return item
|
| 85 |
|
| 86 |
|