Guest User

Untitled

a guest
Jul 21st, 2018
96
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.96 KB | None | 0 0
  1. maindir = 'E:\opticalFlowDataSets\MPI-Sintel-complete\training\flow';
  2. dstdir = 'G:\sintel_result\sp_flow_5_4';
  3. % subdir = dir(maindir);
  4. subdir = dir(dstdir);
  5. aae_sum = 0;
  6. stdae_sum = 0;
  7. aepe_sum = 0;
  8. count = 0;
  9. aae_scene(1) = 0, stdae_scene(1) = 0, aepe_scene(1) = 0;
  10. for i = 1 : length( subdir )
  11. if( isequal( subdir( i ).name, '.' )|| isequal( subdir( i ).name, '..')|| ~subdir( i ).isdir) % 如果不是目录则跳过
  12. continue;
  13. end
  14. sub_dstdir = fullfile(dstdir, subdir(i).name, '*.flo'); % 自己的.flo文件
  15.  
  16. subdirpath = fullfile( maindir, subdir( i ).name, '*.flo' ); % ground truth 文件
  17. gt_path = dir( subdirpath ); % 子文件夹下找后缀为flo的文件
  18. flow_path = dir(sub_dstdir);
  19.  
  20. aae_scene_sum = 0;
  21. stdae_scene_sum = 0;
  22. aepe_scene_sum = 0;
  23. scene_count = 0;
  24. for j = 1 : length( gt_path )
  25. gtFile = fullfile( maindir, subdir( i ).name, gt_path( j ).name);
  26. flowFile = fullfile(dstdir, subdir(i).name, flow_path(j).name);
  27.  
  28. gt = readFlowFile(gtFile);
  29. flow = readFlowFile(flowFile);
  30.  
  31. [aae stdae aepe] = flowAngErr(gt(:,:,1), gt(:,:,2), flow(:,:,1), flow(:,:,2), 0);
  32. aae_scene_sum = aae_scene_sum + aae;
  33. stdae_scene_sum = stdae_scene_sum + stdae;
  34. aepe_scene_sum = aepe_scene_sum + aepe;
  35. count = count + 1;
  36. scene_count = scene_count + 1;
  37. % 此处添加你的对文件读写操作 %
  38. end
  39. aae_sum = aae_sum + aae_scene_sum;
  40. stdae_sum = stdae_sum + stdae_scene_sum;
  41. aepe_sum = aepe_sum + aepe_scene_sum;
  42. aae_scene_sum = aae_scene_sum / scene_count;
  43. stdae_scene_sum = stdae_scene_sum / scene_count;
  44. aepe_scene_sum = aepe_scene_sum / scene_count;
  45. aae_scene(i) = aae_scene_sum;
  46. stdae_scene(i) = stdae_scene_sum;
  47. aepe_scene(i) = aepe_scene_sum;
  48. end
  49. aae_sum = aae_sum / count;
  50. stdae_sum = stdae_sum / count;
  51. aepe_sum = aepe_sum / count;
Add Comment
Please, Sign In to add comment