其他分享
首页 > 其他分享> > detectron2 中的coco数据集默认位置更改

detectron2 中的coco数据集默认位置更改

作者:互联网

detectron2 中的coco数据集默认位置更改

其默认的dataset的位置是 .../detectron2/data/datasets/...

更改方式

打开上述路径下的buildin.py文件

# -*- coding: utf-8 -*-
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved

"""
This file registers pre-defined datasets at hard-coded paths, and their metadata.

We hard-code metadata for common datasets. This will enable:
1. Consistency check when loading the datasets
2. Use models on these standard datasets directly and run demos,
   without having to download the dataset annotations

We hard-code some paths to the dataset that's assumed to
exist in "./datasets/". #其默认位置为"./datasets/"

Users SHOULD NOT use this file to create new dataset / metadata for new dataset.
To add new dataset, refer to the tutorial "docs/DATASETS.md".
"""

更改最下面的函数:

# Register them all under "./datasets"
_root = os.getenv("DETECTRON2_DATASETS", "/media/code/T1/") #更改DATASETS的值
register_all_coco(_root)
register_all_lvis(_root)
register_all_cityscapes(_root)
register_all_pascal_voc(_root)

标签:datasets,更改,detectron2,register,默认,dataset,code,coco,root
来源: https://www.cnblogs.com/jiashun/p/12674998.html