def __init__(self, host='192.168.1.124', port=8117):
print(os.getpid())
self.host = host
self.port = port
self.thread_flage = 0
self.bufsize = 1024 * 3
self.capture_queue = Queue(maxsize=cfg.max_queue_len)
self.result_queue = Queue(maxsize=cfg.max_queue_len)
# self.enable_capture = Event()
# self.enable_predict = Event()
# self.capture_thread = CaptureThread(self.capture_queue, self.enable_capture)
# capture_thread.start()
self.detect_thread = DetectThread(self.capture_queue,
self.result_queue)
self.detect_thread.start()
# self.audio_thread = AudioThread()
# self.audio_thread.start()
self.output_path = "output_%s.mp4" % str(uuid.uuid4())
self.deep_squat = DeepSquat()
self.back_squat = BackSquat()
# self.capture_thread = CaptureThread(self.capture_queue, self.enable_capture)
# capture_thread.start()
self.socket_server()
def __init__(self, host='192.168.1.124', port=8117):
print(os.getpid())
self.host = host
self.port = port
self.bufsize = 14096
self.capture_queue = Queue(maxsize=cfg.max_queue_len)
self.result_queue = Queue(maxsize=cfg.max_queue_len)
self.detect_thread = DetectThread(self.capture_queue,
self.result_queue)
self.detect_thread.start()
self.img_time = {}
self.output_path = "output_%s.mp4" % str(uuid.uuid4())
self.deep_squat = DeepSquat()
self.back_squat = BackSquat()
self.socket_server()
from cfgs.config import cfg
from actions import *
if __name__ == "__main__":
capture_queue = Queue(maxsize=cfg.max_queue_len)
result_queue = Queue(maxsize=cfg.max_queue_len)
enable_capture = Event()
enable_predict = Event()
capture_thread = CaptureThread(capture_queue, enable_capture)
capture_thread.start()
detect_thread = DetectThread(capture_queue, result_queue, enable_capture,
enable_predict)
detect_thread.start()
audio_thread = AudioThread()
audio_thread.start()
output_path = "output_%s.mp4" % str(uuid.uuid4())
deep_squat = DeepSquat()
back_squat = BackSquat()
visualize_gui = VisualizeGUI(result_queue,
audio_thread,
enable_predict,
back_squat,
output_path=output_path)
class ServerAccept:
def __init__(self, host='192.168.1.124', port=8117):
print(os.getpid())
self.host = host
self.port = port
self.bufsize = 14096
self.capture_queue = Queue(maxsize=cfg.max_queue_len)
self.result_queue = Queue(maxsize=cfg.max_queue_len)
self.detect_thread = DetectThread(self.capture_queue,
self.result_queue)
self.detect_thread.start()
self.img_time = {}
self.output_path = "output_%s.mp4" % str(uuid.uuid4())
self.deep_squat = DeepSquat()
self.back_squat = BackSquat()
self.socket_server()
def receive_data(self, conn, addr):
print(os.getpid())
while True:
print(os.getpid())
receive_time = time.time()
buf = b""
data_len = -1
# pdb.set_trace()
# receive one image
while True:
tem_buf = conn.recv(self.bufsize)
buf += tem_buf
if data_len != -1 and data_len == len(buf):
break
if len(buf) > 4 and data_len == -1:
img_size = struct.unpack('i', buf[:4])
data_len = img_size[0] + 16
# pdb.set_trace()
img_info = struct.unpack("iid%ds" % (data_len - 16), buf)
print("received ori time ", str(img_info[2]))
img_id = img_info[1]
self.img_time[img_id] = img_info[2]
if img_id == -1:
self.capture_queue.put([addr, -1, 0])
conn.close()
break
else:
data = np.fromstring(img_info[3], dtype='uint8')
img = cv2.imdecode(data, 1)
print("======received from client ======", addr, img_id,
img.shape, str(time.time() - img_info[2]))
print("size ", self.capture_queue.qsize())
self.capture_queue.put([addr, img_id, img])
print("Client connection interrupted: {0}".format(addr))
conn.close()
self.s.close()
print("{0} closed! ".format(addr))
def send_result(self, addr, port):
try:
send_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
send_client.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
except Exception as e:
print("Error create socket")
print("111")
try:
# socket.setdefaulttimeout(5)
print("server to client(addr:port)", addr)
send_client.connect((addr, 8122))
except socket.error as e:
print("Error connecting to client: %s" % e)
sys.exit()
while True:
frame_id, tip, result_img = self.result_queue.get()
if frame_id == -1:
send_client.close()
break
send_time = time.time()
img_encode = cv2.imencode('.jpg', result_img)[1]
img_code = np.array(img_encode)
str_encode = img_code.tostring()
struc_2 = "ii%ds" % (len(str_encode))
# print(struc_2)
# data2 = struct.pack(struc_2, int(len(str_encode)), int(frame_id), str_encode)
# tip = ['tips/back_squat/good_tip_1.wav', 'tips/back_squat/good_tip_2.wav']
tem_tip = ""
if len(tip) >= 2:
tem_tip = "-".join(e for e in tip)
elif len(tip) == 1:
tem_tip = tip[0]
else:
continue
# data2 = struct.pack("%ds" % (len(tip)))
result = pickle.dumps((tem_tip))
send_client.send(result)
print("%d peak send over, pid %d, total deal time:" %
(frame_id, os.getpid()))
def socket_server(self):
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
self.s.bind((self.host, self.port))
except socket.error as e:
print("Bind failed")
print(e)
sys.exit()
self.s.listen(5)
print("waiting for connection...")
print(os.getpid())
def deal_data(self):
conn, addr = self.s.accept()
print("accept new connection from {0}".format(addr))
t = threading.Thread(target=self.receive_data, args=(conn, addr))
t.start()
# self.wait_for_result()
# self.receive_data(conn, addr)
t1 = threading.Thread(target=self.send_result, args=(addr))
t1.start()
def wait_for_result(self):
while True:
if self.thread_flage:
self.receive_data()
class ServerAccept:
def __init__(self, host='192.168.1.124', port=8117):
print(os.getpid())
self.host = host
self.port = port
self.bufsize = 1024 * 3
self.capture_queue = Queue(maxsize=cfg.max_queue_len)
self.result_queue = Queue(maxsize=cfg.max_queue_len)
self.detect_thread = DetectThread(self.capture_queue,
self.result_queue)
self.detect_thread.start()
self.output_path = "output_%s.mp4" % str(uuid.uuid4())
self.deep_squat = DeepSquat()
self.back_squat = BackSquat()
self.socket_server()
def receive_data(self, conn, addr):
print(os.getpid())
while True:
print(os.getpid())
receive_time = time.time()
buf = b""
data_len = -1
# receive one image
while True:
tem_buf = conn.recv(self.bufsize)
buf += tem_buf
if data_len != -1 and data_len == len(buf):
break
if len(buf) > 4 and data_len == -1:
img_size = struct.unpack('i', buf[:4])
data_len = img_size[0] + 8
img_info = struct.unpack("i%ds" % (data_len - 8), buf[4:])
print("receive time ", str(time.time() - receive_time))
img_id = img_info[0]
if img_id == -1:
self.capture_queue.put([addr, -1, 0])
conn.close()
break
else:
data = np.fromstring(img_info[1], dtype='uint8')
img = cv2.imdecode(data, 1)
print("======received from client ======", addr, img_id,
img.shape)
self.capture_queue.put([addr, img_id, img])
print("Client connection interrupted: {0}".format(addr))
conn.close()
self.s.close()
print("{0} closed! ".format(addr))
def send_result(self, addr, port):
try:
send_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
send_client.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
except Exception as e:
print("Error create socket")
print("111")
try:
# socket.setdefaulttimeout(5)
print("server to client(addr:port)", addr)
send_client.connect((addr, 8121))
except socket.error as e:
print("Error connecting to client: %s" % e)
sys.exit()
while True:
client_addr, ids, peak = self.result_queue.get()
if ids == -1:
send_client.close()
break
result = pickle.dumps((ids, peak))
# print("server send to client", addr, client_addr)
# print(ids, peak)
send_client.send(result)
print("%d peak send over, pid %d" % (ids, os.getpid()))
def socket_server(self):
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
self.s.bind((self.host, self.port))
except socket.error as e:
print("Bind failed")
print(e)
sys.exit()
self.s.listen(5)
print("waiting for connection...")
print(os.getpid())
def deal_data(self):
while True:
conn, addr = self.s.accept()
print("accept new connection from {0}".format(addr))
t = threading.Thread(target=self.receive_data, args=(conn, addr))
t.start()
# self.wait_for_result()
# self.receive_data(conn, addr)
t1 = threading.Thread(target=self.send_result, args=(addr))
t1.start()
def wait_for_result(self):
while True:
if self.thread_flage:
self.receive_data()
class Server_Accept:
def __init__(self, host='192.168.1.124', port=8117):
print(os.getpid())
self.host = host
self.port = port
self.thread_flage = 0
self.bufsize = 1024 * 3
self.capture_queue = Queue(maxsize=cfg.max_queue_len)
self.result_queue = Queue(maxsize=cfg.max_queue_len)
# self.enable_capture = Event()
# self.enable_predict = Event()
# self.capture_thread = CaptureThread(self.capture_queue, self.enable_capture)
# capture_thread.start()
self.detect_thread = DetectThread(self.capture_queue,
self.result_queue)
self.detect_thread.start()
# self.audio_thread = AudioThread()
# self.audio_thread.start()
self.output_path = "output_%s.mp4" % str(uuid.uuid4())
self.deep_squat = DeepSquat()
self.back_squat = BackSquat()
# self.capture_thread = CaptureThread(self.capture_queue, self.enable_capture)
# capture_thread.start()
self.socket_server()
# visualize_gui = VisualizeGUI(result_queue, audio_thread, enable_predict, back_squat, output_path=output_path)
def del_data(self, conn, addr):
print(os.getpid())
# .from_camera = video_file == None
while 1:
print(os.getpid())
buf = b""
data_len = -1
while True:
tem_buf = conn.recv(self.bufsize)
buf += tem_buf
if data_len != -1 and data_len == len(buf):
break
if len(buf) > 4 and data_len == -1:
img_size = struct.unpack('i', buf[:4])
data_len = img_size[0] + 16
img_info = struct.unpack("iid%ds" % (data_len - 16), buf)
print("received ori time ", str(img_info[2]))
img_id = img_info[1]
# self.img_time[img_id] = img_info[2]
if img_id == -1:
self.capture_queue.put([addr, -1, 0])
conn.close()
break
else:
data = np.fromstring(img_info[3], dtype='uint8')
img = cv2.imdecode(data, 1)
print("======received from client ======", addr, img_id,
img.shape, str(time.time() - img_info[2]))
print("size ", self.capture_queue.qsize())
self.capture_queue.put([addr, img_id, img])
client_addr, ids, peak = self.result_queue.get()
result = pickle.dumps((ids, peak, time.time()))
print("server send to client")
conn.send(result)
print("peak send over")
# break
print("Client connection interrupted: {0}".format(addr))
# conn.close()
# self.s.close()
print("{0} closed! ".format(addr))
def socket_server(self):
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
self.s.bind((self.host, self.port))
except socket.error as e:
print("Bind failed")
print(e)
sys.exit()
self.s.listen(5)
print("waiting for connection...")
def receive_data(self):
# while True:
conn, addr = self.s.accept()
print("accept new connection from {0}".format(addr))
# t = threading.Thread(target=self.del_data, args=(conn, addr))
# t.start()
# self.wait_for_result()
self.del_data(conn, addr)
conn.close()
self.s.close()
def wait_for_result(self):
while True:
if self.thread_flage:
self.receive_data()