Merge pull request #1026 from TeslaZhao/develop

fix pipeline bugs of py3.6

python/examples/ocr/rec_web_server.py

@@ -43,25 +43,21 @@ class OCRService(WebService):
             data = np.fromstring(data, np.uint8)
             im = cv2.imdecode(data, cv2.IMREAD_COLOR)
             img_list.append(im)
-        feed_list = []
         max_wh_ratio = 0
         for i, boximg in enumerate(img_list):
             h, w = boximg.shape[0:2]
             wh_ratio = w * 1.0 / h
             max_wh_ratio = max(max_wh_ratio, wh_ratio)
-        for img in img_list:
+        _, w, h = self.ocr_reader.resize_norm_img(img_list[0],
+                                                  max_wh_ratio).shape
+        imgs = np.zeros((len(img_list), 3, w, h)).astype('float32')
+        for i, img in enumerate(img_list):
             norm_img = self.ocr_reader.resize_norm_img(img, max_wh_ratio)
-            #feed = {"image": norm_img}
-            feed_list.append(norm_img)
-        if len(feed_list) == 1:
-            feed_batch = {
-                "image": np.concatenate(
-                    feed_list, axis=0)[np.newaxis, :]
-            }
-        else:
-            feed_batch = {"image": np.concatenate(feed_list, axis=0)}
+            imgs[i] = norm_img
+
+        feed = {"image": imgs.copy()}
         fetch = ["ctc_greedy_decoder_0.tmp_0", "softmax_0.tmp_0"]
-        return feed_batch, fetch, True
+        return feed, fetch, True
 
     def postprocess(self, feed={}, fetch=[], fetch_map=None):
         rec_res = self.ocr_reader.postprocess(fetch_map, with_score=True)

python/pipeline/channel.py

@@ -326,7 +326,10 @@ class ProcessChannel(object):
             with self._cv:
                 while self._stop.value == 0:
                     try:
-                        self._que.put({op_name: channeldata}, timeout=0)
+                        self._que.put((channeldata.id, {
+                            op_name: channeldata
+                        }),
+                                      timeout=0)
                         break
                     except Queue.Full:
                         self._cv.wait()
@@ -378,7 +381,7 @@ class ProcessChannel(object):
             else:
                 while self._stop.value == 0:
                     try:
-                        self._que.put(put_data, timeout=0)
+                        self._que.put((data_id, put_data), timeout=0)
                         break
                     except Queue.Empty:
                         self._cv.wait()
@@ -414,7 +417,7 @@ class ProcessChannel(object):
             with self._cv:
                 while self._stop.value == 0 and resp is None:
                     try:
-                        resp = self._que.get(timeout=0)
+                        resp = self._que.get(timeout=0)[1]
                         break
                     except Queue.Empty:
                         if timeout is not None:
@@ -459,7 +462,7 @@ class ProcessChannel(object):
             while self._stop.value == 0 and self._consumer_cursors[
                     op_name] - self._base_cursor.value >= len(self._output_buf):
                 try:
-                    channeldata = self._que.get(timeout=0)
+                    channeldata = self._que.get(timeout=0)[1]
                     self._output_buf.append(channeldata)
                     list_values = list(channeldata.values())
                     _LOGGER.debug(
@@ -633,7 +636,10 @@ class ThreadChannel(Queue.PriorityQueue):
             with self._cv:
                 while self._stop is False:
                     try:
-                        self.put({op_name: channeldata}, timeout=0)
+                        self.put((channeldata.id, {
+                            op_name: channeldata
+                        }),
+                                 timeout=0)
                         break
                     except Queue.Full:
                         self._cv.wait()
@@ -680,7 +686,7 @@ class ThreadChannel(Queue.PriorityQueue):
             else:
                 while self._stop is False:
                     try:
-                        self.put(put_data, timeout=0)
+                        self.put((data_id, put_data), timeout=0)
                         break
                     except Queue.Empty:
                         self._cv.wait()
@@ -716,7 +722,7 @@ class ThreadChannel(Queue.PriorityQueue):
             with self._cv:
                 while self._stop is False and resp is None:
                     try:
-                        resp = self.get(timeout=0)
+                        resp = self.get(timeout=0)[1]
                         break
                     except Queue.Empty:
                         if timeout is not None:

python/pipeline/profiler.py

@@ -120,7 +120,7 @@ class PerformanceTracer(object):
                                                          tot_cost))
 
             if "DAG" in op_cost:
-                calls = op_cost["DAG"].values()
+                calls = list(op_cost["DAG"].values())
                 calls.sort()
                 tot = len(calls)
                 qps = 1.0 * tot / self._interval_s