rwightman / efficientdet-pytorch

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

Is there typo? #7

Closed pgsrv closed 4 years ago

pgsrv commented 4 years ago

https://github.com/rwightman/pytorch-image-models/blob/master/timm/data/config.py#L82 in function:

def get_mean_by_model(model_name):
    model_name = model_name.lower()
    if 'dpn' in model_name:
        return IMAGENET_DPN_STD
    elif 'ception' in model_name or ('nasnet' in model_name and 'mnasnet' not in model_name):
        return IMAGENET_INCEPTION_MEAN
    else:
        return IMAGENET_DEFAULT_MEAN

expected IMAGENET_DPN_MEAN?

pgsrv commented 4 years ago

Sorry, wrong repo