gathierry/FastFlow

Can I/How to replace the resnet18 feature extractor by a MobileNetV3_large backbone?

Closed this issue · 1 comments

Can I/How to replace the resnet18 feature extractor by a MobileNetV3_large backbone?

As this repo uses pretrained backbones from timm, you can simply replace the backbone_name=resnet18 with backbone_name=mobilenetv3_large_100 in config file.

You also need to extend this constant to get rid of the assertion error

SUPPORTED_BACKBONES = [