32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
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 |