Spaces:
Running
on
Zero
Running
on
Zero
Update difpoint/src/utils/cropper.py
Browse files
difpoint/src/utils/cropper.py
CHANGED
@@ -48,7 +48,7 @@ class Cropper(object):
|
|
48 |
device = "cuda"
|
49 |
face_analysis_wrapper_provicer = ["CUDAExecutionProvider"]
|
50 |
self.landmark_runner = LandmarkRunner(
|
51 |
-
ckpt_path=
|
52 |
onnx_provider=device,
|
53 |
device_id=device_id,
|
54 |
)
|
|
|
48 |
device = "cuda"
|
49 |
face_analysis_wrapper_provicer = ["CUDAExecutionProvider"]
|
50 |
self.landmark_runner = LandmarkRunner(
|
51 |
+
ckpt_path=self.crop_cfg.landmark_ckpt_path,
|
52 |
onnx_provider=device,
|
53 |
device_id=device_id,
|
54 |
)
|