sequenceExtract.py 文件源码

def plotParams(self, th_sigma_high, th_weight_low, size, title = ''):
        colors = ['r', 'y', 'g', 'c', 'm', 'b']
        paramNames = ['Weight', 'CenterX', 'CenterY', 'SigmaX', 'SigmaY']
        low_lims = [th_weight_low, 0, 0, 0,0]
        high_lims = [None, size, size, th_sigma_high, th_sigma_high]
        fig1 = pylab.figure(figsize = (15,15))
        parameters2, parameters3 = [], []
        for i in xrange(self.lengthSeq):
            for obj in self.listObj[i]:
                if obj.nComponents == 2:
                    parameters2.extend(obj.params)
                elif obj.nComponents == 3:
                    parameters3.extend(obj.params)
        for nParam in range(5):
            pylab.subplot(3,2,nParam+1)
            pylab.plot(range(self.lengthSeq), np.abs(parameters2[nParam::12]), 'b-')
            pylab.plot(range(self.lengthSeq), np.abs(parameters2[6+nParam::12]), 'm-')
            pylab.axhline(y = low_lims[nParam], color = 'r', ls = '-', lw = 5)
            if high_lims[nParam] != 0:
                pylab.axhline(y = high_lims[nParam], color = 'r', ls = '-', lw = 5)
            pylab.title(paramNames[nParam])
        pylab.savefig("{0}_params2.png".format(title))
        pylab.close()
        fig1 = pylab.figure(figsize = (20,15))
        for nParam in range(5):
            pylab.subplot(3,2,nParam+1)
            pylab.plot(range(self.lengthSeq), np.abs(parameters3[nParam::18]), 'b-')
            pylab.plot(range(self.lengthSeq), np.abs(parameters3[6+nParam::18]), 'm-')
            pylab.plot(range(self.lengthSeq), np.abs(parameters3[12+nParam::18]), 'g-')
            pylab.axhline(y = low_lims[nParam], color = 'r', ls = '-', lw = 5)
            if high_lims[nParam] != 0:
                pylab.axhline(y = high_lims[nParam], color = 'r', ls = '-', lw = 5)
            pylab.title(paramNames[nParam])
        pylab.savefig("{0}_params3.png".format(title))
        pylab.close()