Update datasets/INFERENCE_demo.py

datasets/INFERENCE_demo.py

@@ -118,7 +118,7 @@ class INFERENCE_demo(torch.utils.data.Dataset):
             for i, score in enumerate(scores):
                 if score < self.score_threshold:
                     break
-                if i>self.num_person:
+                if i>=self.num_person:
                     break
                 save_name = img_paths[ann_idx].split('/')[-1]
                 save_name = save_name.split('.')[0]