dk - edit

2026-07-09 18:16:06 +09:00
parent dda1759322
commit 4f1c320b3b
25 changed files with 1744 additions and 4 deletions
@@ -0,0 +1,32 @@
+import os
+from .reiddataset_downloader import*
+def import_DukeMTMC(dataset_dir):
+    dukemtmc_dir = os.path.join(dataset_dir, 'DukeMTMC-reID')
+    if not os.path.exists(dukemtmc_dir):
+        print('Please Download the DukMTMC Dataset')
+    data_group = ['train','query','gallery']
+    for group in data_group:
+        if group == 'train':
+            name_dir = os.path.join(dukemtmc_dir , 'bounding_box_train')
+        elif group == 'query':
+            name_dir = os.path.join(dukemtmc_dir, 'query')
+        else:
+            name_dir = os.path.join(dukemtmc_dir, 'bounding_box_test')
+        file_list=os.listdir(name_dir)
+        globals()[group]={}
+        for name in file_list:
+            if name[-3:]=='jpg':
+                id = name.split('_')[0]
+                if id not in globals()[group]:
+                    globals()[group][id]=[]
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])            
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])
+                    globals()[group][id].append([])
+                cam_n = int(name.split('_')[1][1])-1
+                globals()[group][id][cam_n].append(os.path.join(name_dir,name))
+    return train, query, gallery