We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bee95bd commit 04bfe34Copy full SHA for 04bfe34
README.md
@@ -12,7 +12,7 @@ from face_detector import YoloDetector
12
import numpy as np
13
from PIL import Image
14
15
-model = YoloDetector(target_size=720,gpu=0,min_face=90)
+model = YoloDetector(target_size=720, device="cuda:0", min_face=90)
16
orgimg = np.array(Image.open('test_image.jpg'))
17
bboxes,points = model.predict(orgimg)
18
```
0 commit comments