Flag name inconsistency

jerryli27 · Aug 9, 2020 · 718e1cb · 718e1cb
1 parent 34484d7
commit 718e1cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/infer_from_image.py b/infer_from_image.py
@@ -97,7 +97,7 @@ def main(_):
       feed_dict={image_ph: image_np})
 
     if FLAGS.output_cropped_image:
-      if FLAGS.only_output_cropped_single_person and len(result["detection_score"]) == 1:
+      if FLAGS.only_output_cropped_single_object and len(result["detection_score"]) == 1:
         num_outputs = 1
       else:
         num_outputs = len(result["detection_score"])