7eb23e3c1d
* Initial commit * Initial commit * Cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix precommit errors * Remove TF builds from CI * export last.pt * Created using Colaboratory * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
import argparse
|
|
|
|
from wandb_utils import WandbLogger
|
|
|
|
from utils.general import LOGGER
|
|
|
|
WANDB_ARTIFACT_PREFIX = 'wandb-artifact://'
|
|
|
|
|
|
def create_dataset_artifact(opt):
|
|
logger = WandbLogger(opt, None, job_type='Dataset Creation') # TODO: return value unused
|
|
if not logger.wandb:
|
|
LOGGER.info("install wandb using `pip install wandb` to log the dataset")
|
|
|
|
|
|
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='YOLOv3', help='name of W&B Project')
|
|
parser.add_argument('--entity', default=None, help='W&B entity')
|
|
parser.add_argument('--name', type=str, default='log dataset', help='name of W&B run')
|
|
|
|
opt = parser.parse_args()
|
|
opt.resume = False # Explicitly disallow resume check for dataset upload job
|
|
|
|
create_dataset_artifact(opt)
|