perf(dataset): 修改原图和标注文件的读取路径

Author: zjZSTU

py/lib/models/location_dataset.py

@@ -39,8 +39,8 @@ def __init__(self, root_dir, cate_list, transform=None, S=7, B=2, C=20):
         xml_path_list = []
         for name in cate_list:
             for i in range(1, 61):
-                jpeg_path_list.append(os.path.join(root_dir, '%s_%d.jpg' % (name, i)))
-                xml_path_list.append(os.path.join(root_dir, '%s_%d.xml' % (name, i)))
+                jpeg_path_list.append(os.path.join(root_dir, 'imgs', '%s_%d.jpg' % (name, i)))
+                xml_path_list.append(os.path.join(root_dir, 'annotations', '%s_%d.xml' % (name, i)))
         self.jpeg_path_list = jpeg_path_list
         self.xml_path_list = xml_path_list
 
@@ -128,7 +128,7 @@ def __len__(self):
 
 
 if __name__ == '__main__':
-    root_dir = '../../data/training_images/'
+    root_dir = '../../data/location_dataset/'
     transform = transforms.Compose([
         transforms.ToPILImage(),
         transforms.Resize((448, 448)),