Skip to content

Commit 4fafc47

Browse files
fix param
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent 4857208 commit 4fafc47

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/anomalib/models/image/rkde/region_extractor.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(
4545

4646
self.backbone = maskrcnn_resnet50_fpn_v2(
4747
weights=MaskRCNN_ResNet50_FPN_V2_Weights.DEFAULT,
48-
rpn_post_nms_top_n_test=max_detections_per_image,
48+
rpn_post_nms_top_n_test=1000,
4949
rpn_score_thresh=0.0,
5050
box_score_thresh=score_threshold,
5151
rpn_nms_thresh=0.3,

0 commit comments

Comments
 (0)