AI_Learning_Assistants/backheatsamba.py at main · 2021BUSTER/AI_Learning_Assistants · 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
import serial
import numpy as np

import matplotlib.pyplot as plt
import seaborn as sns

PORT = '/dev/ttyUSB1'
BaudRate = 9600

ARD = serial.Serial(PORT,BaudRate)

array=[]
buf=[]

cnt = 0

def Ardread(cnt): # return list [Ard1,Ard2]

    if ARD.readable():
        LINE = ARD.readline()
        data = LINE.decode('utf-8')
        data = data.strip('\n')
        data = data.strip('\r')
        splitData = data.split(',')

        for i in range(0,9) :
            splitData[i] = int(splitData[i])

        splitData = np.reshape(splitData,(3,3))

        ax = sns.heatmap(splitData, cmap='Blues', cbar=False , vmin = 0, vmax = 1000)
        ax.tick_params(left=False, bottom=False)
        ax.axes.xaxis.set_visible(False)
        ax.axes.yaxis.set_visible(False)

        fname = 'back' + str(cnt) +'.png'
        plt.savefig(fname, dpi=200)

    else :
        print("읽기 실패 from _Ardread_")

while (True):
    Ardread(cnt)
    cnt += 1