Update README.md

This commit is contained in:
xuehao.ma 2022-07-14 21:15:57 +08:00 committed by GitHub
parent 20c209a5ba
commit 80e55e9869
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -123,26 +123,26 @@ Intel|i7-8700(X86-cpu)|Linux(amd64)|ncnn|4.51ms|4.33ms
```
### Build the training .yaml configuration file
* Reference./configs/coco.yaml
```
DATASET:
TRAIN: "/home/qiuqiu/Desktop/coco2017/train2017.txt" # Train dataset path .txt file
VAL: "/home/qiuqiu/Desktop/coco2017/val2017.txt" # Val dataset path .txt file
NAMES: "dataset/coco128/coco.names" # .names category label file
MODEL:
NC: 80 # Number of detection categories
INPUT_WIDTH: 352 # The width of the model input image
INPUT_HEIGHT: 352 # The height of the model input image
TRAIN:
LR: 0.001 # Train learn rate
THRESH: 0.25 #
WARMUP: true # Trun on warm up
BATCH_SIZE: 64 # Batch size
END_EPOCH: 350 # Train epichs
MILESTIONES: # Declining learning rate steps
- 150
- 250
- 300
```
```
DATASET:
TRAIN: "/home/qiuqiu/Desktop/coco2017/train2017.txt" # Train dataset path .txt file
VAL: "/home/qiuqiu/Desktop/coco2017/val2017.txt" # Val dataset path .txt file
NAMES: "dataset/coco128/coco.names" # .names category label file
MODEL:
NC: 80 # Number of detection categories
INPUT_WIDTH: 352 # The width of the model input image
INPUT_HEIGHT: 352 # The height of the model input image
TRAIN:
LR: 0.001 # Train learn rate
THRESH: 0.25 #
WARMUP: true # Trun on warm up
BATCH_SIZE: 64 # Batch size
END_EPOCH: 350 # Train epichs
MILESTIONES: # Declining learning rate steps
- 150
- 250
- 300
```
### Train
* Perform training tasks
```
@ -203,12 +203,15 @@ TRAIN:
pip install onnx-runtime
python3 runtime.py
```
@misc{=FastestDet,
title={FastestDet: Ultra lightweight anchor-free real-time object detection algorithm.},
author={xuehao.ma},
howpublished = {\url{https://github.com/dog-qiuqiu/FastestDet}},
year={2022}
}
```
# Citation
* If you find this project useful in your research, please consider cite:
```
@misc{=FastestDet,
title={FastestDet: Ultra lightweight anchor-free real-time object detection algorithm.},
author={xuehao.ma},
howpublished = {\url{https://github.com/dog-qiuqiu/FastestDet}},
year={2022}
}
```
# Reference
* https://github.com/Tencent/ncnn