diff --git a/preprocessor_config.json b/preprocessor_config.json index 6ff682d..ce34b46 100644 --- a/preprocessor_config.json +++ b/preprocessor_config.json @@ -1,7 +1,7 @@ { "do_normalize": true, "do_resize": true, - "feature_extractor_type": "DetrFeatureExtractor", + "feature_extractor_type": "YolosFeatureExtractor", "format": "coco_detection", "image_mean": [ 0.485,