Colab/GPEN not working

[Orchoidizer]

on - !python face_enhancement.py

Traceback (most recent call last): File "face_enhancement.py", line 90, in faceenhancer = FaceEnhancement(size=model['size'], model=model['name'], channel_multiplier=2) File "face_enhancement.py", line 19, in init self.facedetector = RetinaFaceDetection(base_dir) File "/content/GPEN/retinaface/retinaface_detection.py", line 26, in init self.load_model() File "/content/GPEN/retinaface/retinaface_detection.py", line 47, in load_model pretrained_dict = torch.load(self.pretrained_path, map_location=lambda storage, loc: storage.cuda()) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 581, in load with _open_file_like(f, 'rb') as opened_file: File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 230, in _open_file_like return _open_file(name_or_buffer, mode) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 211, in init super(_open_file, self).init(open(name, mode)) FileNotFoundError: [Errno 2] No such file or directory: './weights/RetinaFace-R50.pth'

[yangxy]

Please update the colab file.

[BillTribble]

Same problem here!

[zndomination]

Hi, any update? it doesnt work

[yvoonne06]

Download the models on the main page.