NanjingForestryUniversity/supermachine-wood
3
0
Fork 0
mirror of https://github.com/NanjingForestryUniversity/supermachine-wood.git synced 2025-11-08 18:23:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加的训练模型代码

Browse Source
This commit is contained in:
FEIJINTI 2022-09-15 17:49:23 +08:00
parent e53268ec72
commit e327c538b0
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
socket_detector.py
View File

@ -18,8 +18,7 @@ def main():
# receive data # receive data
t1 = time.time() t1 = time.time()
size_buff = socket_receive.recv(5) size_buff = socket_receive.recv(5)
if size_buff[4] == 0: if (size_buff[4] == 0):
n_rows, n_cols = size_buff[0] << 8 | size_buff[1], size_buff[2] << 8 | size_buff[3] n_rows, n_cols = size_buff[0] << 8 | size_buff[1], size_buff[2] << 8 | size_buff[3]
data_size = n_rows * n_cols * 3 data_size = n_rows * n_cols * 3
print(data_size) print(data_size)
@ -48,6 +47,15 @@ def main():
socket_send.send(b'Q') socket_send.send(b'Q')
print('Q send success') print('Q send success')
print((time.time()-t1)) print((time.time()-t1))
elif (size_buff[4] == 1):
detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
detector.correct()
detector.fit_pictures(data_path=r"C:\Users\FEIJINTI\PycharmProjects\wood_color")
socket_send.send(b'G')
elif (size_buff[4] == 2):
pass