Skip to content

Commit 04bfe34

Browse files
authored
fixed function parameters
1 parent bee95bd commit 04bfe34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ from face_detector import YoloDetector
1212
import numpy as np
1313
from PIL import Image
1414

15-
model = YoloDetector(target_size=720,gpu=0,min_face=90)
15+
model = YoloDetector(target_size=720, device="cuda:0", min_face=90)
1616
orgimg = np.array(Image.open('test_image.jpg'))
1717
bboxes,points = model.predict(orgimg)
1818
```

0 commit comments

Comments
 (0)