Spaces:
Runtime error
Runtime error
File size: 2,460 Bytes
d7dbcdd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import os
import os.path
import glob
def dataloader(filepath, level=6):
iml0 = []
iml1 = []
flowl0 = []
disp0 = []
dispc = []
calib = []
level_stars = '/*'*level
candidate_pool = glob.glob('%s/optical_flow%s'%(filepath,level_stars))
for flow_path in sorted(candidate_pool):
if 'TEST' in flow_path: continue
if 'flower_storm_x2/into_future/right/OpticalFlowIntoFuture_0023_R.pfm' in flow_path:
continue
if 'flower_storm_x2/into_future/left/OpticalFlowIntoFuture_0023_L.pfm' in flow_path:
continue
if 'flower_storm_augmented0_x2/into_future/right/OpticalFlowIntoFuture_0023_R.pfm' in flow_path:
continue
if 'flower_storm_augmented0_x2/into_future/left/OpticalFlowIntoFuture_0023_L.pfm' in flow_path:
continue
if 'FlyingThings' in flow_path and '_0014_' in flow_path:
continue
if 'FlyingThings' in flow_path and '_0015_' in flow_path:
continue
idd = flow_path.split('/')[-1].split('_')[-2]
if 'into_future' in flow_path:
idd_p1 = '%04d'%(int(idd)+1)
else:
idd_p1 = '%04d'%(int(idd)-1)
if os.path.exists(flow_path.replace(idd,idd_p1)):
d0_path = flow_path.replace('/into_future/','/').replace('/into_past/','/').replace('optical_flow','disparity')
d0_path = '%s/%s.pfm'%(d0_path.rsplit('/',1)[0],idd)
dc_path = flow_path.replace('optical_flow','disparity_change')
dc_path = '%s/%s.pfm'%(dc_path.rsplit('/',1)[0],idd)
im_path = flow_path.replace('/into_future/','/').replace('/into_past/','/').replace('optical_flow','frames_cleanpass')
im0_path = '%s/%s.png'%(im_path.rsplit('/',1)[0],idd)
im1_path = '%s/%s.png'%(im_path.rsplit('/',1)[0],idd_p1)
#with open('%s/camera_data.txt'%(im0_path.replace('frames_cleanpass','camera_data').rsplit('/',2)[0]),'r') as f:
# if 'FlyingThings' in flow_path and len(f.readlines())!=40:
# print(flow_path)
# continue
iml0.append(im0_path)
iml1.append(im1_path)
flowl0.append(flow_path)
disp0.append(d0_path)
dispc.append(dc_path)
calib.append('%s/camera_data.txt'%(im0_path.replace('frames_cleanpass','camera_data').rsplit('/',2)[0]))
return iml0, iml1, flowl0, disp0, dispc, calib
|