修改了一个bug

This commit is contained in:
FEIJINTI 2022-09-16 10:19:30 +08:00
parent ba41005b24
commit 5a674a7400

View File

@ -55,11 +55,10 @@ def main():
socket_send.send(b'G') socket_send.send(b'G')
elif (size_buff[4] == 2): elif (size_buff[4] == 2):
str_size = size_buff[0] << 8 | size_buff[1] str_size = size_buff[0] << 8 | size_buff[1]
select_model = socket_receive.recv(str_size) select_model = socket_receive.recv(str_size).decode("ascii")
model_path = os.path.join("models", select_model) model_path = os.path.join("models", select_model)
detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False) detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
detector.load(path=model_path) detector.load(path=model_path)
break