caffe_python_layers.py 文件源码

def reshape(self, bottom, top):
        while self.redis.llen(config.solver_prototxt) == 0:
            print 'waiting for spider...', self.redis.llen(config.solver_prototxt)
            time.sleep(0.2)

        self.item = cPickle.loads(self.redis.lpop(config.solver_prototxt))
        item = self.item

        for i in range(len(item)):
            top[i].reshape(*item[i].shape)