X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=example-apps%2FPDD%2Fpcb-defect-detection%2Flibs%2Fexport_pbs%2Ftest_exportPb.py;fp=example-apps%2FPDD%2Fpcb-defect-detection%2Flibs%2Fexport_pbs%2Ftest_exportPb.py;h=0000000000000000000000000000000000000000;hb=3ed2c61d9d7e7916481650c41bfe5604f7db22e9;hp=ad78e42289bd7bfece75fa0fc3e7907db38d594a;hpb=e6d40ddb2640f434a9d7d7ed99566e5e8fa60cc1;p=ealt-edge.git

diff --git a/example-apps/PDD/pcb-defect-detection/libs/export_pbs/test_exportPb.py b/example-apps/PDD/pcb-defect-detection/libs/export_pbs/test_exportPb.py
deleted file mode 100755
index ad78e42..0000000
--- a/example-apps/PDD/pcb-defect-detection/libs/export_pbs/test_exportPb.py
+++ /dev/null
@@ -1,81 +0,0 @@
-# -*- coding:utf-8 -*-
-
-from __future__ import absolute_import
-from __future__ import print_function
-from __future__ import division
-
-import os, sys
-import tensorflow as tf
-import time
-import cv2
-import argparse
-import numpy as np
-sys.path.append("../")
-
-from data.io.image_preprocess import short_side_resize_for_inference_data
-from libs.configs import cfgs
-from libs.networks import build_whole_network
-from libs.box_utils import draw_box_in_img
-from help_utils import tools
-
-
-
-
-
-def load_graph(frozen_graph_file):
-
-    # we parse the graph_def file
-    with tf.gfile.GFile(frozen_graph_file, 'rb') as f:
-        graph_def = tf.GraphDef()
-        graph_def.ParseFromString(f.read())
-
-    # we load the graph_def in the default graph
-
-    with tf.Graph().as_default() as graph:
-        tf.import_graph_def(graph_def,
-                            input_map=None,
-                            return_elements=None,
-                            name="",
-                            op_dict=None,
-                            producer_op_list=None)
-    return graph
-
-
-def test(frozen_graph_path, test_dir):
-
-    graph = load_graph(frozen_graph_path)
-    print("we are testing ====>>>>", frozen_graph_path)
-
-    img = graph.get_tensor_by_name("input_img:0")
-    dets = graph.get_tensor_by_name("DetResults:0")
-
-    with tf.Session(graph=graph) as sess:
-        for img_path in os.listdir(test_dir):
-            a_img = cv2.imread(os.path.join(test_dir, img_path))[:, :, ::-1]
-            st = time.time()
-            dets_val = sess.run(dets, feed_dict={img: a_img})
-
-            show_indices = dets_val[:, 1] >= 0.5
-            dets_val = dets_val[show_indices]
-            final_detections = draw_box_in_img.draw_boxes_with_label_and_scores(a_img,
-                                                                                boxes=dets_val[:, 2:],
-                                                                                labels=dets_val[:, 0],
-                                                                                scores=dets_val[:, 1])
-            cv2.imwrite(img_path,
-                        final_detections[:, :, ::-1])
-            print ("%s cost time: %f" % (img_path, time.time() - st))
-
-if __name__ == '__main__':
-    test('/home/yjr/PycharmProjects/Faster-RCNN_Tensorflow/output/Pbs/FasterRCNN_Res101_Pascal_Frozen.pb',
-         '/home/yjr/PycharmProjects/Faster-RCNN_Tensorflow/tools/demos')
-
-
-
-
-
-
-
-
-
-
-