dataset.py 文件源码

python
阅读 351 收藏 0 点赞 0 评论 0

项目:crnn 作者: wulivicte 项目源码 文件源码
def __init__(self, root=None, transform=None, target_transform=None):
        self.env = lmdb.open(
            root,
            max_readers=1,
            readonly=True,
            lock=False,
            readahead=False,
            meminit=False)

        if not self.env:
            print('cannot creat lmdb from %s' % (root))
            sys.exit(0)

        with self.env.begin(write=False) as txn:
            nSamples = int(txn.get('num-samples'))
            self.nSamples = nSamples

        self.transform = transform
        self.target_transform = target_transform
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号