fix(dataset): 图像和标注文件一一对应

Author: zjZSTU

py/lib/models/location_dataset.py

@@ -36,8 +36,13 @@ def __init__(self, root_dir, cate_list, transform=None, S=7, B=2, C=20):
         self.C = C
         self.cate_list = cate_list
 
-        self.jpeg_path_list = glob.glob(os.path.join(root_dir, 'imgs', '*.jpg'))
-        self.xml_path_list = glob.glob(os.path.join(root_dir, 'annotations', '*.xml'))
+        jpeg_path_list = glob.glob(os.path.join(root_dir, 'imgs', '*.jpg'))
+        jpeg_path_list.sort()
+        xml_path_list = glob.glob(os.path.join(root_dir, 'annotations', '*.xml'))
+        xml_path_list.sort()
+
+        self.jpeg_path_list = jpeg_path_list
+        self.xml_path_list = xml_path_list
 
     def __getitem__(self, index):
         """