def process(self): self.socket.send_string(root.art_style_filename, zmq.SNDMORE) self.socket.setsockopt(zmq.RCVTIMEO, 15000) self.socket.setsockopt(zmq.SNDTIMEO, 15000) try: send_array(self.socket, root.photo_content) except: self.done("網路連線失敗") return while 1: try: img = recv_array(self.socket) print("recv") except: img = np.array([]) self.done("網路連線太慢") print("recv timeout") return if img.shape == (0, ): print("done") break print("ok") try: self.socket.send_string("ok") except: print("send timeout") self.done("網路連線超時") return self.update_progress(img) self.done()
def process(self): self.socket.send_string(root.art_style_filename, zmq.SNDMORE) self.socket.setsockopt(zmq.RCVTIMEO, 15000) self.socket.setsockopt(zmq.SNDTIMEO, 15000) try: send_array(self.socket, root.photo_content) except: self.done("網路連線失敗") return while 1: try: img = recv_array(self.socket) print("recv") except: img = np.array([]) self.done("網路連線太慢") print("recv timeout") return if img.shape == (0,): print("done") break print("ok") try: self.socket.send_string("ok") except: print("send timeout") self.done("網路連線超時") return self.update_progress(img) self.done()
def server(): context = zmq.Context() socket = context.socket(zmq.REP) socket.bind ("tcp://*:7788") i = 0 while 1: print("start loop") socket.setsockopt(zmq.RCVTIMEO, -1) art_style_filename = socket.recv_string() print(art_style_filename) socket.setsockopt(zmq.RCVTIMEO, 10000) socket.setsockopt(zmq.SNDTIMEO, 1000) photo_content = recv_array(socket) print("get photo_content", photo_content.shape) for img in neuralStyle.p_transfer(photo_content, art_style_filename): send_array(socket, img) print("send") s = socket.recv_string() print("ok", s) send_array(socket, np.array([])) print("done")