14 lines
278 B
Python
14 lines
278 B
Python
#!/usr/bin/env python
|
|
|
|
import cv2
|
|
|
|
from core.detectors import CornerNet_Saccade
|
|
from core.vis_utils import draw_bboxes
|
|
|
|
detector = CornerNet_Saccade()
|
|
image = cv2.imread("demo.jpg")
|
|
|
|
bboxes = detector(image)
|
|
image = draw_bboxes(image, bboxes)
|
|
cv2.imwrite("demo_out.jpg", image)
|