| 4 | import random |
| 5 | |
| 6 | def process_query_sysu(data_path, mode = 'all', relabel=False): |
| 7 | if mode== 'all': |
| 8 | ir_cameras = ['cam3','cam6'] |
| 9 | elif mode =='indoor': |
| 10 | ir_cameras = ['cam3','cam6'] |
| 11 | |
| 12 | file_path = os.path.join(data_path,'exp/test_id.txt') |
| 13 | files_rgb = [] |
| 14 | files_ir = [] |
| 15 | |
| 16 | with open(file_path, 'r') as file: |
| 17 | ids = file.read().splitlines() |
| 18 | ids = [int(y) for y in ids[0].split(',')] |
| 19 | ids = ["%04d" % x for x in ids] |
| 20 | |
| 21 | for id in sorted(ids): |
| 22 | for cam in ir_cameras: |
| 23 | img_dir = os.path.join(data_path,cam,id) |
| 24 | if os.path.isdir(img_dir): |
| 25 | new_files = sorted([img_dir+'/'+i for i in os.listdir(img_dir)]) |
| 26 | files_ir.extend(new_files) |
| 27 | query_img = [] |
| 28 | query_id = [] |
| 29 | query_cam = [] |
| 30 | for img_path in files_ir: |
| 31 | camid, pid = int(img_path[-15]), int(img_path[-13:-9]) |
| 32 | query_img.append(img_path) |
| 33 | query_id.append(pid) |
| 34 | query_cam.append(camid) |
| 35 | return query_img, np.array(query_id), np.array(query_cam) |
| 36 | |
| 37 | def process_gallery_sysu(data_path, mode = 'all', trial = 0, relabel=False): |
| 38 | |