Spaces:

mrneuralnet
/

P-DFD

Sleeping

mrneuralnet commited on Aug 8, 2023

Commit

e10b61a

1 Parent(s): 25981df

Change to cpu

Files changed (1) hide show

extract_video.py CHANGED Viewed

@@ -226,7 +226,7 @@ if __name__ == '__main__':
     # net and model
     net = RetinaFace(cfg=cfg, phase='test')
-    net = load_model(net, pretrained_weights, load_to_cpu=False)
     net.eval()
     print('Finished loading model!')

     # net and model
     net = RetinaFace(cfg=cfg, phase='test')
+    net = load_model(net, pretrained_weights, load_to_cpu=False if device=='cuda' else True)
     net.eval()
     print('Finished loading model!')