scorer-python/saveimg.py at master · futurestandard/scorer-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import cv2
import sys
import scorer_sdk.Scorer
import pycurl

def send2line_on():
    buffer='on.png saved'
    curl = pycurl.Curl()
    curl.setopt(pycurl.URL, 'https://notify-api.line.me/api/notify')
    curl.setopt(pycurl.HTTPHEADER, ['Authorization: Bearer AhfffLL3f1BYUJMnBnV8HZFAuvFS3ztJ65ZQv9UPo5j'])
    curl.setopt(pycurl.HTTPPOST, [('message', buffer.encode('utf-8')),('imageFile', (curl.FORM_FILE, '/opt/scorer/home/dev/SDK/roi/on.png'))])
    curl.perform()

def send2line_off():
    buffer='off.png saved'
    curl = pycurl.Curl()
    curl.setopt(pycurl.URL, 'https://notify-api.line.me/api/notify')
    curl.setopt(pycurl.HTTPHEADER, ['Authorization: Bearer AhfffLL3f1BYUJMnBnV8HZFAuvFS3ztJ65ZQv9UPo5j'])
    curl.setopt(pycurl.HTTPPOST, [('message', buffer.encode('utf-8')),('imageFile', (curl.FORM_FILE, '/opt/scorer/home/dev/SDK/roi/off.png'))])
    curl.perform()

args = sys.argv
scorer=scorer_sdk.Scorer.Scorer("Task")
x1,y1,x2,y2 = scorer.get_roi_rect_by_index(0)


if len(args) >1:
    if scorer.poll() == False:
        sys.exit()

    frame = scorer.get_frame()

    if frame == None:
        sys.exit()

    bgr = frame.get_bgr()
    rect = bgr[y1:y2, x1:x2]

    if args[1] == "on":
        #cv2.imwrite("/opt/scorer/home/dev/scorer-python/on.png",rect)
        cv2.imwrite("/opt/scorer/home/dev/SDK/roi/on.png",rect)
        send2line_on()
        print("on saved")

    elif args[1] == "off":
        cv2.imwrite("/opt/scorer/home/dev/SDK/roi/off.png",rect)
        send2line_off()
        print("off saved")