mirror of
https://github.com/NanjingForestryUniversity/supermachine-wood.git
synced 2025-11-08 18:23:54 +00:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import socket
|
|
import numpy as np
|
|
import cv2
|
|
from classifer import WoodClass
|
|
import time
|
|
import os
|
|
from utils import PreSocket, process_cmd
|
|
|
|
|
|
|
|
def main():
|
|
socket_receive = PreSocket(socket.AF_INET, socket.SOCK_STREAM)
|
|
socket_receive.connect(('127.0.0.1', 21122))
|
|
socket_send = PreSocket(socket.AF_INET, socket.SOCK_STREAM)
|
|
socket_send.connect(('127.0.0.1', 21123))
|
|
process_cmd(recv_sock=socket_receive , send_sock=socket_send)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 2个端口
|
|
# 接受端口21122
|
|
# 发送端口21123
|
|
# 接收到图片 n_rows * n_bands * n_cols, float32
|
|
# 发送图片 n_rows * n_cols, uint8
|
|
main()
|
|
# test(r"D:\build-tobacco-Desktop_Qt_5_9_0_MSVC2015_64bit-Release\calibrated15.raw")
|
|
# main()
|
|
# debug_main()
|
|
# test_run(all_data_dir=r'D:\数据')
|
|
# with open(r'D:\数据\虫子\valid2.raw', 'rb') as f:
|
|
# data = np.frombuffer(f.read(), dtype=np.float32).reshape(600, 29, 1024).transpose(0, 2, 1)
|
|
# plt.matshow(data[:, :, 10])
|
|
# plt.show()
|
|
# detector = SpecDetector('model_spec/model_29.p')
|
|
# result = detector.predict(data)
|
|
#
|
|
# plt.matshow(result)
|
|
# plt.show()
|
|
# result = result.reshape((600, 1024))
|