Update log_dataset.py

ultralytics · Feb 5, 2021 · 9e5e9be · 9e5e9be
1 parent b12b20b
commit 9e5e9be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/wandb_logging/log_dataset.py b/utils/wandb_logging/log_dataset.py
@@ -15,8 +15,8 @@ def create_dataset_artifact(opt):
     logger = WandbLogger(opt, '', None, data, job_type='create_dataset')
     nc, names = (1, ['item']) if opt.single_cls else (int(data['nc']), data['names'])
     names = {k: v for k, v in enumerate(names)}  # to index dictionary
-    logger.log_dataset_artifact(LoadImagesAndLabels(data['train'], data_type="train"), names, name='train')  # trainset
-    logger.log_dataset_artifact(LoadImagesAndLabels(data['val'], data_type="val"), names, name='val')  # valset
+    logger.log_dataset_artifact(LoadImagesAndLabels(data['train']), names, name='train')  # trainset
+    logger.log_dataset_artifact(LoadImagesAndLabels(data['val']), names, name='val')  # valset
 
     # Update data.yaml with artifact links
     data['train'] = WANDB_ARTIFACT_PREFIX + str(Path(opt.project) / 'train')