Merge pull request #224 from computational-cell-analytics/u-import

constantinpape · web-flow · commit eb331b9da161 · 2023-10-04T09:51:11.000+02:00
Merge fixes into master
diff --git a/micro_sam/evaluation/inference.py b/micro_sam/evaluation/inference.py
@@ -94,7 +94,7 @@ def _get_batched_prompts(
     # apply the transforms to the points and boxes
     if use_boxes:
         input_boxes = torch.from_numpy(
-            transform_function.apply_boxes(input_points.numpy(), gt.shape)
+            transform_function.apply_boxes(input_boxes.numpy(), gt.shape)
         )
     if use_points:
         input_points = torch.from_numpy(

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def _get_batched_prompts(`
`94`	`94`	`# apply the transforms to the points and boxes`
`95`	`95`	`if use_boxes:`
`96`	`96`	`input_boxes = torch.from_numpy(`
`97`		`- transform_function.apply_boxes(input_points.numpy(), gt.shape)`
	`97`	`+ transform_function.apply_boxes(input_boxes.numpy(), gt.shape)`
`98`	`98`	`)`
`99`	`99`	`if use_points:`
`100`	`100`	`input_points = torch.from_numpy(`