Spaces:
Runtime error
Runtime error
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')
|