Glenn Jocher
efe60b5681
* Update loggers * Config * Update val.py * cleanup * fix1 * fix2 * fix3 and reformat * format sweep.py * Logger() class * cleanup * cleanup2 * wandb package import fix * wandb package import fix2 * txt fix * fix4 * fix5 * fix6 * drop wandb into utils/loggers * fix 7 * rename loggers/wandb_logging to loggers/wandb * Update message * Update message * Update message * cleanup * Fix x axis bug * fix rank 0 issue * cleanup
27 lines
870 B
Python
27 lines
870 B
Python
import argparse
|
|
|
|
import yaml
|
|
|
|
from wandb_utils import WandbLogger
|
|
|
|
WANDB_ARTIFACT_PREFIX = 'wandb-artifact://'
|
|
|
|
|
|
def create_dataset_artifact(opt):
|
|
with open(opt.data) as f:
|
|
data = yaml.safe_load(f) # data dict
|
|
logger = WandbLogger(opt, '', None, data, job_type='Dataset Creation')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--data', type=str, default='data/coco128.yaml', help='data.yaml path')
|
|
parser.add_argument('--single-cls', action='store_true', help='train as single-class dataset')
|
|
parser.add_argument('--project', type=str, default='YOLOv5', help='name of W&B Project')
|
|
parser.add_argument('--entity', default=None, help='W&B entity')
|
|
|
|
opt = parser.parse_args()
|
|
opt.resume = False # Explicitly disallow resume check for dataset upload job
|
|
|
|
create_dataset_artifact(opt)
|