rwightman / efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Apache License 2.0
1.58k stars 293 forks source link

Updated model_config.py #129

Closed ghnreigns closed 3 years ago

ghnreigns commented 3 years ago

Changed image_size = 1280 to image_size=(1280, 1280). It will not be affected if you override the config's image size by saying config['image_size'] = (512,512) for example but it might be affected if you stick to default.