Skip to content

Commit e3421ed

Browse files
committed
[fix] bug - criticial for pandaset use case
1 parent 35fe604 commit e3421ed

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

compressai_vision/model_wrappers/detectron2.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,11 @@ def __init__(self, device="cpu", **kwargs):
611611

612612
@torch.no_grad()
613613
def _feature_pyramid_to_output(
614-
self, x: Dict, org_img_size: Dict, input_img_size: List
614+
self,
615+
x: Dict,
616+
org_img_size: Dict,
617+
input_img_size: List,
618+
hyper_params: Optional[Dict] = None,
615619
):
616620
"""
617621
performs downstream task using the feature pyramid ['p2', 'p3', 'p4', 'p5']

0 commit comments

Comments
 (0)