Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- maindir = 'E:\opticalFlowDataSets\MPI-Sintel-complete\training\flow';
- dstdir = 'G:\sintel_result\sp_flow_5_4';
- % subdir = dir(maindir);
- subdir = dir(dstdir);
- aae_sum = 0;
- stdae_sum = 0;
- aepe_sum = 0;
- count = 0;
- aae_scene(1) = 0, stdae_scene(1) = 0, aepe_scene(1) = 0;
- for i = 1 : length( subdir )
- if( isequal( subdir( i ).name, '.' )|| isequal( subdir( i ).name, '..')|| ~subdir( i ).isdir) % 如果不是目录则跳过
- continue;
- end
- sub_dstdir = fullfile(dstdir, subdir(i).name, '*.flo'); % 自己的.flo文件
- subdirpath = fullfile( maindir, subdir( i ).name, '*.flo' ); % ground truth 文件
- gt_path = dir( subdirpath ); % 子文件夹下找后缀为flo的文件
- flow_path = dir(sub_dstdir);
- aae_scene_sum = 0;
- stdae_scene_sum = 0;
- aepe_scene_sum = 0;
- scene_count = 0;
- for j = 1 : length( gt_path )
- gtFile = fullfile( maindir, subdir( i ).name, gt_path( j ).name);
- flowFile = fullfile(dstdir, subdir(i).name, flow_path(j).name);
- gt = readFlowFile(gtFile);
- flow = readFlowFile(flowFile);
- [aae stdae aepe] = flowAngErr(gt(:,:,1), gt(:,:,2), flow(:,:,1), flow(:,:,2), 0);
- aae_scene_sum = aae_scene_sum + aae;
- stdae_scene_sum = stdae_scene_sum + stdae;
- aepe_scene_sum = aepe_scene_sum + aepe;
- count = count + 1;
- scene_count = scene_count + 1;
- % 此处添加你的对文件读写操作 %
- end
- aae_sum = aae_sum + aae_scene_sum;
- stdae_sum = stdae_sum + stdae_scene_sum;
- aepe_sum = aepe_sum + aepe_scene_sum;
- aae_scene_sum = aae_scene_sum / scene_count;
- stdae_scene_sum = stdae_scene_sum / scene_count;
- aepe_scene_sum = aepe_scene_sum / scene_count;
- aae_scene(i) = aae_scene_sum;
- stdae_scene(i) = stdae_scene_sum;
- aepe_scene(i) = aepe_scene_sum;
- end
- aae_sum = aae_sum / count;
- stdae_sum = stdae_sum / count;
- aepe_sum = aepe_sum / count;
Add Comment
Please, Sign In to add comment