Update default parameters

zhiqwang · Nov 26, 2020 · ef3981a · ef3981a
1 parent 4480999
commit ef3981a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -52,7 +52,7 @@ conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
 To read a source image and detect its objects run:
 
 ```bash
-python -m detect [--model_cfg ./models/yolov5s.yaml]
+python -m detect [--model_cfg yolov5s.yaml]
                  [--input_source YOUR_IMAGE_SOURCE_DIR]
                  [--labelmap ./notebooks/assets/coco.names]
                  [--output_dir ./data-bin/output]

diff --git a/detect.py b/detect.py
@@ -132,7 +132,7 @@ def main(args):
     import argparse
     parser = argparse.ArgumentParser(description=__doc__)
 
-    parser.add_argument('--model_cfg', type=str, default='./models/yolov5s.yaml',
+    parser.add_argument('--model_cfg', type=str, default='yolov5s.yaml',
                         help='path where the model cfg in')
     parser.add_argument('--labelmap', type=str, default='./notebooks/assets/coco.names',
                         help='path where the coco category in')