SSD Small SSDMobilenetV2 - wom-ai/inference_results_v0.5 GitHub Wiki

Setup

$ diff ./SSDMobileNet.py ./SSDMobileNet_for_ssd_mobilenet_v2_coco_2018_03_29.py 
61c61,62
<     inputOrder=[0, 7, 6],
---
>     #inputOrder=[0, 7, 6],
>     inputOrder=[6, 0, 12],
71c72,73
<     "MultipleGridAnchorGenerator/Concatenate": concat_priorbox,
---
>     #"MultipleGridAnchorGenerator/Concatenate": concat_priorbox,
>     "Concatenate": concat_priorbox,
98a101,102
>     print('=======================================================================')
>     print("TEST 1", dynamic_graph.find_nodes_by_op("NMS_OPT_TRT"))
103c107,111
<     dynamic_graph.find_nodes_by_name("Input")[0].input.remove("image_tensor:0")
---
>     #print("TEST", dynamic_graph.find_nodes_by_name("Input")[0].input)
>     #dynamic_graph.find_nodes_by_name("Input")[0].input.remove("image_tensor:0")
> 
>     #dynamic_graph.remove(dynamic_graph.find_nodes_by_op("GridAnchor_TRT"))
>     #dynamic_graph.remove(dynamic_graph.find_nodes_by_op("NMS_OPT_TRT"))
108a117
>         print(args)
111c120,122
<         self.model_path = dict_get(args, "model_path", default="build/models/SSDMobileNet/frozen_inference_graph.pb")
---
>         #self.model_path = dict_get(args, "model_path", default="build/models/SSDMobileNet/frozen_inference_graph.pb")
>         self.model_path = dict_get(args, "model_path", default="build/models/ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb")
>         #self.model_path = dict_get(args, "model_path", default="build/models/ssd_inception_v2_coco_2018_01_28/frozen_inference_graph.pb")
137a149
>         #parser.register_output("MultipleGridAnchorGenerator")