コード例 #1
0
 def VizGrayFlows(self, key, flowList):
     imGrayList = self.flowVizsGray[key + 'Vizg']
     assert len(flowList) == len(imGrayList)
     if (len(flowList) == 0):
         print('WARRING VizGrayFlows: Not find any flowfiles')
     for flowName, imGrayName in zip(flowList, imGrayList):
         flow = read_gen(flowName)
         flow_g = abs_flow(flow)
         imsave(imGrayName, flow_g / np.max(flow_g))
コード例 #2
0
# -*- coding: utf-8 -*-
"""
Created on Tue Dec  4 16:31:24 2018

@author: Administrator
"""
import sys
from os.path import join
from scipy.misc import imsave,imread
from PIL import Image
import numpy as np

sys.path.append("../Server_EasyTest")
from myflowlib import read_gen,abs_flow


path= 'E:\我的文档\任务\配准1\测试图片\展示在文档中的\z3'
name = 'z3res.flo'
save_name ='z3res_gray.jpg'

name = join(path,name)
save_name = join(path,save_name)

res = read_gen(name)
res = abs_flow(res)
imsave(save_name,res/np.max(res))
コード例 #3
0
    def SparplotSimple(self, ids, netout_flow, uncertainty_flow,
                       groundtruth_flow):
        gt = groundtruth_flow
        res = abs_flow(uncertainty_flow)
        flow = netout_flow
        assert flow.shape == uncertainty_flow.shape == gt.shape
        best = abs_flow(gt - flow)
        total_steps = self.spar_steps

        aepe0 = EPE(flow, gt)
        print('AEPE:' + str(aepe0))
        totalpixels = int(flow.size / 2)
        remainpixels = np.linspace(totalpixels,
                                   0,
                                   total_steps,
                                   endpoint=False,
                                   dtype='int')
        res_sort = res.flatten()
        best_sort = best.flatten()
        res_sort_index = np.argsort(res_sort)
        best_sort_index = np.argsort(best_sort)

        res_aepe = []
        res_threshold = []
        for p in remainpixels:
            threshold_id = res_sort_index[p - 1]
            threshold = res_sort[threshold_id]
            aepe = EPE_usingmask(flow, gt, res < threshold)
            res_aepe.append(aepe)
            res_threshold.append(threshold)
            print(u"\rFinish" + str(int(len(res_aepe) / total_steps * 50)) +
                  '%',
                  end='')

        best_aepe = []
        best_threshold = []
        for p in remainpixels:
            threshold_id = best_sort_index[p - 1]
            threshold = best_sort[threshold_id]
            aepe = EPE_usingmask(flow, gt, best < threshold)
            best_aepe.append(aepe)
            best_threshold.append(threshold)
            print(u"\rFinish" +
                  str(int(len(best_aepe) / total_steps * 50 + 50)) + '%',
                  end='')

        x = (totalpixels - remainpixels) / totalpixels
        y1 = res_aepe / aepe0
        y2 = best_aepe / aepe0

        plt.plot(x, y1, mec='r', mfc='w', label='Our-Net')
        plt.plot(x, y2, ms=10, label='Oracle')
        plt.legend()  # 让图例生效
        plt.margins(0)
        plt.xlabel('Fraction of Removed Pixels')  #X轴标签
        plt.ylabel('Average EPE (Normalized)')  #Y轴标签
        plt.title('Sparsification Plots')  #标题

        plt.savefig(self.spars['sparPlot'][ids])
        plt.close()
        save_3ziplist(self.spars['sparData'][ids],zip(x,y1,y2),\
                      'aepe0:'+str(aepe0)+'\nRemoved\tPreAEPE\tOraAEPE\n')
        flow_write(self.flows['bestFlow'][ids], gt - flow)
        print('\t\tSave fig, data, imgs and bestFlow!')
        return (remainpixels, res_aepe, best_aepe, aepe0)
コード例 #4
0
    def SparplotSimple(self,ids,netout_flow,uncertainty_flow,groundtruth_flow):

        gt = groundtruth_flow
        res = abs_flow(uncertainty_flow)
        flow = netout_flow
        assert flow.shape==uncertainty_flow.shape==gt.shape
        best = abs_flow(gt - flow)
        total_steps=self.spar_steps

        aepe0=EPE(flow,gt)
        print('AEPE:'+str(aepe0))
        totalpixels=int(flow.size/2)
        remainpixels=np.linspace(totalpixels,0,total_steps,endpoint=False,dtype='int')
        res_sort=res.flatten()
        best_sort=best.flatten()
        res_sort_index=np.argsort(res_sort)
        best_sort_index=np.argsort(best_sort)

        res_aepe=[]
        res_threshold=[]
        for p in remainpixels:
            threshold_id=res_sort_index[p-1]
            threshold=res_sort[threshold_id]
            aepe=EPE_usingmask(flow,gt,res<threshold)
            res_aepe.append(aepe)
            res_threshold.append(threshold)
            print( u"\rFinish"+str(int(len(res_aepe)/total_steps*50))+'%',end = '')

        best_aepe=[]
        best_threshold=[]
        for p in remainpixels:
            threshold_id=best_sort_index[p-1]
            threshold=best_sort[threshold_id]
            aepe=EPE_usingmask(flow,gt,best<threshold)
            best_aepe.append(aepe)
            best_threshold.append(threshold)
            print( u"\rFinish"+str(int(len(best_aepe)/total_steps*50+50))+'%',end = '')

        x=(totalpixels-remainpixels)/totalpixels
        y1=res_aepe/aepe0
        y2=best_aepe/aepe0

        plt.plot(x, y1, mec='r', mfc='w',label='Pred-Merged')
        plt.plot(x, y2, ms=10,label='Oracle')
        plt.legend()  # 让图例生效
        plt.margins(0)
        plt.xlabel('Fraction of Removed Pixels') #X轴标签
        plt.ylabel('Average EPE (Normalized)') #Y轴标签
        plt.title('Sparsification Plots') #标题
        plt.savefig(self.spar_plotlist[ids])
        plt.close() 
        save_3ziplist(self.spar_datalist[ids],zip(x,y1,y2),\
                      'aepe0:'+str(aepe0)+'\nRemoved\tPreAEPE\tOraAEPE\n')
        print('\t\tSave fig, data and imgs !')
        if(self.spar_style=='gray' or self.spar_style=='default'):
            # 1 通道
            absgt=abs_flow(gt)
            imsave(self.spar_grayreslist[ids], res/np.max(res))
            imsave(self.spar_graybestlist[ids], best/np.max(best))
            imsave(self.spar_graygtlist[ids], absgt/np.max(absgt))
        if(self.spar_style=='color' or self.spar_style=='default'):
            # 2通道
            bestflow=Image.fromarray(viz_flow(gt - flow))
            imsave(self.spar_bestlist[ids], bestflow)

        return (remainpixels,res_aepe,best_aepe,aepe0)