acnazarejr / fdet

An easy to use face detection module based on MTCNN and RetinaFace algorithms.
MIT License
19 stars 4 forks source link

url for MobileNet state dict is not working #15

Open Nithin745 opened 3 years ago

Nithin745 commented 3 years ago
def _load_mobile_net_model(self):
        model = _MobileNetV1()
        state_dict = load_state_dict_from_url(
            'https://www.dropbox.com/s/bd1keyo085pscfu/mobilenetv1_pretrain.pt?dl=1',
            map_location=self._device_control
        )
        # # load params
        model.load_state_dict(state_dict)
        return model``

this dropbox link for mobile net is not working