videoFeatureExtractor_lasagne.py 文件源码

def _get_stblock(self, data_input, hnd, mdlt, start_frame=None):
        goodness = False
        if start_frame is None:
                start_frame = random.randint(0, len(data_input['min_length'])-self.step*(self.nframes-1)-1)
        stblock = numpy.zeros([self.nframes, self.block_size, self.block_size])
        for ii in xrange(self.nframes):
                v = data_input[hnd][mdlt][start_frame + ii * self.step]
                mm = abs(numpy.ma.maximum(v))
                if mm > 0.:
                        # normalize to zero mean, unit variance,
                        # concatenate in spatio-temporal blocks
                        stblock[ii] = self.prenormalize(v)
                        goodness = True
        return stblock, goodness