Cleanup model implementation

Cleans up model implementation to:

• Re-use all backbones from torchvision
• Simplify checkpointer
• Remove own implementation of model_zoo
• Implement normalization for torchvision-based backbones
• Enable pytorch installation on osx

bob/ip/binseg/configs/models/driu.py

@@ -11,9 +11,8 @@ Reference: [MANINIS-2016]_
 """
 
 from torch.optim.lr_scheduler import MultiStepLR
-from bob.ip.binseg.modeling.driu import build_driu
-from bob.ip.binseg.utils.model_zoo import modelurls
-from bob.ip.binseg.modeling.losses import SoftJaccardBCELogitsLoss
+from bob.ip.binseg.models.driu import driu
+from bob.ip.binseg.models.losses import SoftJaccardBCELogitsLoss
 from bob.ip.binseg.engine.adabound import AdaBound
 
 ##### Config #####
@@ -29,9 +28,7 @@ amsbound = False
 scheduler_milestones = [900]
 scheduler_gamma = 0.1
 
-model = build_driu()
-
-pretrained_backbone = modelurls["vgg16"]
+model = driu()
 
 optimizer = AdaBound(
     model.parameters(),