MCPcopy
hub / github.com/black0017/MedicalZooPytorch / find_random_crop_dim

Function find_random_crop_dim

lib/medloaders/medical_loader_utils.py:203–223  ·  view source on GitHub ↗
(full_vol_dim, crop_size)

Source from the content-addressed store, hash-verified

201
202
203def find_random_crop_dim(full_vol_dim, crop_size):
204 assert full_vol_dim[0] >= crop_size[0], "crop size is too big"
205 assert full_vol_dim[1] >= crop_size[1], "crop size is too big"
206 assert full_vol_dim[2] >= crop_size[2], "crop size is too big"
207
208 if full_vol_dim[0] == crop_size[0]:
209 slices = crop_size[0]
210 else:
211 slices = np.random.randint(full_vol_dim[0] - crop_size[0])
212
213 if full_vol_dim[1] == crop_size[1]:
214 w_crop = crop_size[1]
215 else:
216 w_crop = np.random.randint(full_vol_dim[1] - crop_size[1])
217
218 if full_vol_dim[2] == crop_size[2]:
219 h_crop = crop_size[2]
220 else:
221 h_crop = np.random.randint(full_vol_dim[2] - crop_size[2])
222
223 return (slices, w_crop, h_crop)
224
225
226def find3Dlabel_boundaries(segmentation_map):

Callers 1

create_sub_volumesFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected