Spaces:
Build error
Build error
Update demo.py
Browse files
demo.py
CHANGED
@@ -32,8 +32,8 @@ MASK_ANNOTATOR = sv.MaskAnnotator()
|
|
32 |
|
33 |
def parse_args():
|
34 |
parser = argparse.ArgumentParser(description='YOLO-World Demo')
|
35 |
-
parser.add_argument('config', help='test config file path')
|
36 |
-
parser.add_argument('checkpoint', help='checkpoint file')
|
37 |
parser.add_argument(
|
38 |
'--work-dir',
|
39 |
help='the directory to save the file containing evaluation metrics',
|
|
|
32 |
|
33 |
def parse_args():
|
34 |
parser = argparse.ArgumentParser(description='YOLO-World Demo')
|
35 |
+
parser.add_argument('config', help='test config file path', default='configs/pretrain/yolo_world_v2_s_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
|
36 |
+
parser.add_argument('checkpoint', help='checkpoint file', default='yolo_world_v2_s_obj365v1_goldg_pretrain-55b943ea.pth')
|
37 |
parser.add_argument(
|
38 |
'--work-dir',
|
39 |
help='the directory to save the file containing evaluation metrics',
|