Model parameters

[Demonicltu]

Hi,

can I somehow change max_det_per_image value? By default it is 100, but for me 1 or 3 is enough. I tried to change it but nothing happened. Maybe somebody faced same issue.

My code:
from effdet import create_model_from_config, get_efficientdet_config

base_config = get_efficientdet_config('tf_efficientdet_d5')
base_config.image_size = (128, 128)
base_config.num_classes=3
base_config.max_det_per_image=3
net = create_model_from_config(base_config, bench_task='predict', num_classes=3)

Kudos for those who will help :D