Spaces:
Configuration error
Configuration error
File size: 235 Bytes
09cc77a |
1 2 3 4 5 |
import torch
import sys
sys.path.insert(0,'./facelib/detection/yolov5face')
model = torch.load('facelib/detection/yolov5face/yolov5n-face.pt', map_location='cpu')['model']
torch.save(model.state_dict(),'../../models/facedetection') |