Merge pull request #10 from ZKTKZ/master

Flag name inconsistency
jerryli27 · Aug 9, 2020 · 98feddd · 98feddd
2 parents 34484d7 + 718e1cb
commit 98feddd
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"])