Fix coco style bounding box annotations (#116)

* Fix coco style bounding box annotations

* Fix top right corner

yolo/tools/data_loader.py

@@ -126,10 +126,12 @@ class YoloDataset(Dataset):
 
     def load_valid_labels(self, label_path: str, seg_data_one_img: list) -> Union[Tensor, None]:
         """
-        Loads and validates bounding box data is [0, 1] from a label file.
+        Loads valid COCO style segmentation data (values between [0, 1]) and converts it to bounding box coordinates
+        by finding the minimum and maximum x and y values.
 
         Parameters:
-            label_path (str): The filepath to the label file containing bounding box data.
+            label_path (str): The filepath to the label file containing annotation data.
+            seg_data_one_img (list): The actual list of annotations (in segmentation format)
 
         Returns:
             Tensor or None: A tensor of all valid bounding boxes if any are found; otherwise, None.

yolo/utils/dataset_utils.py

@@ -104,7 +104,14 @@ def scale_segmentation(
         if "segmentation" in anno:
             seg_list = [item for sublist in anno["segmentation"] for item in sublist]
         elif "bbox" in anno:
-            seg_list = anno["bbox"]
+            x,y,width,height = anno["bbox"]
+            seg_list = [
+                x, y,  # Top-left corner
+                x + width, y,  # Top-right corner
+                x + width, y + height,  # Bottom-right corner
+                x, y + height  # Bottom-left corner
+            ]
+
         scaled_seg_data = (
             np.array(seg_list).reshape(-1, 2) / [w, h]
         ).tolist()  # make the list group in x, y pairs and scaled with image width, height