285dc85691241532cd976e04a089f409e44d7493,src/detection/tensor/detector.py,Detector,detect_multiple,#Detector#Any#,40

Before Change

            answers = []
            for image in image_array_list:
                predictions = self.sess.run(softmax_tensor, {"DecodeJpeg:0": image})
                predictions = np.squeeze(predictions)
                answer = {}
                for node_id in range(len(predictions)):
                    answer[self.labels[node_id]] = predictions[node_id]
                answers.append(answer)
            return answers

    @staticmethod

After Change

        with tf.device("/gpu:0"):
            softmax_tensor = self.sess.graph.get_tensor_by_name("final_result:0")
            threads = [pool.apply_async(operation, args=(self.sess, softmax_tensor, image_array_list[i],)) for i in
                       range(10)]
            results = []
            for x in threads:
                results.append(x.get())

Instances