| 123456789101112131415161718192021222324 |
- 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')
- opt = parser.parse_args()
- opt.resume = False # Explicitly disallow resume check for dataset upload job
- create_dataset_artifact(opt)
|
