Spaces:
Running
on
T4
Running
on
T4
stevengrove
commited on
Commit
•
a722f87
1
Parent(s):
e413c25
Update demo with XL model
Browse files
app.py
CHANGED
@@ -21,8 +21,8 @@ from tools.demo import demo
|
|
21 |
def parse_args():
|
22 |
parser = argparse.ArgumentParser(
|
23 |
description='YOLO-World Demo')
|
24 |
-
parser.add_argument('--config', default='configs/pretrain/
|
25 |
-
parser.add_argument('--checkpoint', default='
|
26 |
parser.add_argument(
|
27 |
'--work-dir',
|
28 |
help='the directory to save the file containing evaluation metrics')
|
|
|
21 |
def parse_args():
|
22 |
parser = argparse.ArgumentParser(
|
23 |
description='YOLO-World Demo')
|
24 |
+
parser.add_argument('--config', default='configs/pretrain/yolo_world_xl_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
|
25 |
+
parser.add_argument('--checkpoint', default='yolo_world_v2_xl_obj365v1_goldg_cc3mlite_pretrain.pth')
|
26 |
parser.add_argument(
|
27 |
'--work-dir',
|
28 |
help='the directory to save the file containing evaluation metrics')
|