From 142150cd04e18b9fe47723b597a5900d58533f96 Mon Sep 17 00:00:00 2001 From: Masahiro Wada Date: Wed, 31 Jul 2019 20:43:06 +0000 Subject: [PATCH] Improve README.md --- README.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5fbbe0f..b279db1 100644 --- a/README.md +++ b/README.md @@ -19,6 +19,7 @@ To run demo type this in the command line: 1. Download binary file with desired weights: 1. Full weights: `wget https://pjreddie.com/media/files/yolov3.weights` 1. Tiny weights: `wget https://pjreddie.com/media/files/yolov3-tiny.weights` + 1. SPP weights: `wget https://pjreddie.com/media/files/yolov3-spp.weights` 2. Run `python ./convert_weights.py` and `python ./convert_weights_pb.py` 3. Run `python ./demo.py --input_img --output_img --frozen_model ` @@ -33,7 +34,9 @@ To run demo type this in the command line: 1. `NCHW` (gpu only) or `NHWC` 4. `--tiny` 1. Use yolov3-tiny - 5. `--ckpt_file` + 5. `--spp` + 1. Use yolov3-spp + 6. `--ckpt_file` 1. Output checkpoint file 2. convert_weights_pb.py: 1. `--class_names` @@ -44,7 +47,9 @@ To run demo type this in the command line: 1. `NCHW` (gpu only) or `NHWC` 4. `--tiny` 1. Use yolov3-tiny - 5. `--output_graph` + 5. `--spp` + 1. Use yolov3-spp + 6. `--output_graph` 1. Location to write the output .pb graph to 3. demo.py 1. `--class_names` @@ -62,4 +67,4 @@ To run demo type this in the command line: 7. `--iou_threshold` 1. Desired iou threshold 8. `--gpu_memory_fraction` - 1. Fraction of gpu memory to work with \ No newline at end of file + 1. Fraction of gpu memory to work with