loader.py 文件源码

def __init__(self, lmdb_image_datapath, lmdb_label_datapath):
                super(UCF_CC_50, self).__init__()
        self.lmdb_image_datapath = lmdb_image_datapath
        self.lmdb_label_datapath = lmdb_label_datapath
        self.images = []
        self.gts = []
        self.total_patches = 0
        self.limits = []
        self.num_files = 0
        self.file_list = []
        self.env_image = lmdb.open(self.lmdb_image_datapath)
        self.env_label = lmdb.open(self.lmdb_label_datapath)
        self.txn_image = self.env_image.begin()
        self.txn_label = self.env_label.begin()
        self.cursor_image = iter(self.txn_image.cursor())
        self.cursor_label = self.txn_label.cursor()