darknet data文件解析

最新推荐文章于 2022-05-12 09:24:29 发布

TYUT_xiaoming

最新推荐文章于 2022-05-12 09:24:29 发布

阅读量681

点赞数

分类专栏：数据集解析

本文链接：https://blog.csdn.net/TYUT_xiaoming/article/details/103305316

版权

数据集解析专栏收录该内容

9 篇文章 1 订阅

订阅专栏

Python strip() 方法用于移除字符串头尾指定的字符（默认为空格或换行符）或字符序列

注意：该方法只能删除开头或是结尾的字符，不能删除中间部分的字符

data文件

classes=1000
# train
train  = /data/imagenet/imagenet1k.train.list
valid  = /data/imagenet/imagenet1k.valid.list
backup = /home/pjreddie/backup/
labels = data/imagenet.labels.list
names  = data/imagenet.shortnames.list
top=5


def parse_data_config(path):
    """Parses the data configuration file"""
    options = dict()
    options['gpus'] = '0,1,2,3'
    options['num_workers'] = '10'
    with open(path, 'r') as fp:
        lines = fp.readlines()
    for line in lines:
        line = line.strip()
        if line == '' or line.startswith('#'):
            continue
        key, value = line.split('=')
        options[key.strip()] = value.strip()
    return options