csmithxc commited on
Commit
0dc9187
1 Parent(s): 3ebca65

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +2 -2
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',