Skip to content

Commit 0724dfe

Browse files
committed
sort key max to min
Signed-off-by: Behrooz <[email protected]>
1 parent 554ed71 commit 0724dfe

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pathology/multiple_instance_learning/panda_mil_train_evaluate_pytorch_gpu.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def main_worker(gpu, args):
285285
keys=["image"],
286286
patch_size=args.tile_size,
287287
max_num_patches=args.tile_count,
288-
sort_key="max",
288+
sort_key="min",
289289
pad_opts={"constant_values": 255},
290290
),
291291
RandFlipd(keys=["image"], spatial_axis=0, prob=0.5),
@@ -303,8 +303,7 @@ def main_worker(gpu, args):
303303
GridPatchd(
304304
keys=["image"],
305305
patch_size=args.tile_size,
306-
max_num_patches=args.tile_count,
307-
sort_key="max",
306+
sort_key="min",
308307
pad_opts={"constant_values": 255},
309308
),
310309
ScaleIntensityRanged(keys=["image"], a_min=np.float32(255), a_max=np.float32(0)),

0 commit comments

Comments
 (0)