diff --git a/utils/dataloader.py b/utils/dataloader.py index 7c4950d183eb7a5de621aa294649ae6a7531b5df..75d49523758c2d9a1ce57930193b1dc88b8a49c9 100644 --- a/utils/dataloader.py +++ b/utils/dataloader.py @@ -112,7 +112,7 @@ class YoloDataset(Dataset): index = 0 place_x = [0, 0, int(w * min_offset_x), int(w * min_offset_x)] - place_y = [0, int(h * min_offset_y), int(w * min_offset_y), 0] + place_y = [0, int(h * min_offset_y), int(h * min_offset_y), 0] for line in annotation_line: # 每一行进行分割 line_content = line.split()