thatbrguy / Pedestrian-Detection

Pedestrian detection using the TensorFlow Object Detection API. Includes multi GPU parallel processing inference.
358 stars 134 forks source link

Performance issue in the definition of testResizeToRangePreservesStaticSpatialShape, object_detection/core/preprocessor_test.py(P1) #11

Open DLPerf opened 3 years ago

DLPerf commented 3 years ago

Hello, I found a performance issue in the definition of testResizeToRangePreservesStaticSpatialShape, object_detection/core/preprocessor_test.py, tf.shape(out_image) will created repeatedly during program execution, resulting in reduced efficiency. I think in_image, out_image and out_image_shape should be created before the loop in testResizeToRangeWithDynamicSpatialShape.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.