Update app.py
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_s_pretrain_FG_v3.py')
|
25 |
+
parser.add_argument('--checkpoint', default='epoch_97.pth')
|
26 |
parser.add_argument(
|
27 |
'--work-dir',
|
28 |
help='the directory to save the file containing evaluation metrics')
|