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

增加了改变模型路径和训练代码（未测试）

Browse Source
This commit is contained in:
FEIJINTI 2022-09-16 09:33:04 +08:00
parent e327c538b0
commit ba41005b24
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
socket_detector.py
View File

@ -3,6 +3,7 @@ import numpy as np
import cv2 import cv2
from classifer import WoodClass from classifer import WoodClass
import time import time
import os
def main(): def main():
@ -53,7 +54,12 @@ def main():
detector.fit_pictures(data_path=r"C:\Users\FEIJINTI\PycharmProjects\wood_color") detector.fit_pictures(data_path=r"C:\Users\FEIJINTI\PycharmProjects\wood_color")
socket_send.send(b'G') socket_send.send(b'G')
elif (size_buff[4] == 2): elif (size_buff[4] == 2):
pass str_size = size_buff[0] << 8 | size_buff[1]
select_model = socket_receive.recv(str_size)
model_path = os.path.join("models", select_model)
detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
detector.load(path=model_path)
break