Update README.md

README.md

@@ -54,32 +54,32 @@ for image_name in os.listdir(folder_path):
             boxes = result["boxes"]
 
             filtered_data = [(score, label, box) for score, label, box in zip(scores, labels, boxes) if score > 0.5][:2] # จำกัดให้เก็บได้ไม่เกิน 2 กล่อง
-            # for score, label, box in zip(scores, labels, boxes):
-                # if score > 0.5:
-            if len(filtered_data) > 0:
-                detected_any = True
-                    
-            for score, label, box in filtered_data:
-                if label.item() == 0:
-                    label_name = "Pneumonia"
-                elif label.item() == 1:
-                    label_name = "Normal"
-                elif label.item() == 2:
-                    label_name = "Pneumonia_bacteria"
-                else:
-                    label_name = "Pneumonia_virus"
-                
-                xmin, ymin, xmax, ymax = [round(i, 2) for i in box.tolist()]
-                print(f" - Detected {label_name} with score {round(score.item(), 3)} at {xmin, ymin, xmax, ymax}")
-                results_list.append({
-                    "image_name": image_name,
-                    "label": label_name,
-                    "xmin": xmin,
-                    "ymin": ymin,
-                    "xmax": xmax,
-                    "ymax": ymax,
-                    "score": round(score.item(), 3),
-                })
+            for score, label, box in zip(scores, labels, boxes):
+                if score > 0.5:
+                    if len(filtered_data) > 0:
+                        detected_any = True
+                            
+                    for score, label, box in filtered_data:
+                        if label.item() == 0:
+                            label_name = "Pneumonia"
+                        elif label.item() == 1:
+                            label_name = "Normal"
+                        elif label.item() == 2:
+                            label_name = "Pneumonia_bacteria"
+                        else:
+                            label_name = "Pneumonia_virus"
+                        
+                        xmin, ymin, xmax, ymax = [round(i, 2) for i in box.tolist()]
+                        print(f" - Detected {label_name} with score {round(score.item(), 3)} at {xmin, ymin, xmax, ymax}")
+                        results_list.append({
+                            "image_name": image_name,
+                            "label": label_name,
+                            "xmin": xmin,
+                            "ymin": ymin,
+                            "xmax": xmax,
+                            "ymax": ymax,
+                            "score": round(score.item(), 3),
+                        })
 
         if not detected_any:
             print(" - No Detect")