add model

model_segvol_single.py

@@ -76,14 +76,19 @@ class SegVolProcessor():
         item['label'] = gt_voxel_ndarray.astype(np.int32)
 
         # transform
-        return item
+        return item['image'], item['label']
+
+    def zoom_transform(self, ct_npy, gt_npy):
+        item = {
+            'image': ct_npy,
+            'label': gt_npy
+        }
         item = self.transform(item)
         item_zoom_out = self.zoom_out_transform(item)
         item['zoom_out_image'] = item_zoom_out['image']
         item['zoom_out_label'] = item_zoom_out['label']
         return item
 
-
     def generate_box(self, pred_pre, bbox_shift=None):
         meaning_post_label = pred_pre # [h, w, d]
         ones_idx = (meaning_post_label > 0).nonzero(as_tuple=True)