Update README.md

This commit is contained in:
bryandlee 2021-02-23 07:47:03 +09:00 committed by GitHub
parent 8314798e04
commit adc0516db9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -15,7 +15,7 @@ python convert_weights.py
**Inference**
```
python test.py --input_dir [image_folder_path]
python test.py --input_dir [image_folder_path] --device [cpu/cuda]
```