Skip to content

Commit 975edc4

Browse files
committed
Update grid patch
Signed-off-by: Behrooz <[email protected]>
1 parent 7bd2ecb commit 975edc4

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pathology/multiple_instance_learning/panda_mil_train_evaluate_pytorch_gpu.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -282,9 +282,9 @@ def main_worker(gpu, args):
282282
RandGridPatchd(
283283
keys=["image"],
284284
patch_size=(args.tile_size, args.tile_size),
285-
fix_num_patches=args.tile_count,
286-
sort_key="min",
287-
pad_opts={"constant_values": 255},
285+
num_patches=args.tile_count,
286+
sort_fn="min",
287+
constant_values=255,
288288
),
289289
RandFlipd(keys=["image"], spatial_axis=0, prob=0.5),
290290
RandFlipd(keys=["image"], spatial_axis=1, prob=0.5),
@@ -301,8 +301,8 @@ def main_worker(gpu, args):
301301
GridPatchd(
302302
keys=["image"],
303303
patch_size=(args.tile_size, args.tile_size),
304-
sort_key="min",
305-
pad_opts={"constant_values": 255},
304+
sort_fn="min",
305+
constant_values=255,
306306
),
307307
ScaleIntensityRanged(keys=["image"], a_min=np.float32(255), a_max=np.float32(0)),
308308
ToTensord(keys=["image", "label"]),

0 commit comments

Comments
 (0)