Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/hlf20010508/pytorch-yolo-v5-fall
基于yolov5的跌倒检测框架
https://github.com/hlf20010508/pytorch-yolo-v5-fall
Last synced: about 5 hours ago
JSON representation
基于yolov5的跌倒检测框架
- Host: GitHub
- URL: https://github.com/hlf20010508/pytorch-yolo-v5-fall
- Owner: hlf20010508
- License: mit
- Created: 2022-06-10T04:28:07.000Z (over 2 years ago)
- Default Branch: master
- Last Pushed: 2022-07-27T05:51:00.000Z (over 2 years ago)
- Last Synced: 2023-03-04T23:24:19.025Z (over 1 year ago)
- Language: Python
- Homepage:
- Size: 819 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
# pytorch-yolo-v5-fall
训练
```
python3 train.py --weights yolov5x.pt --data data/mydata.yaml --epochs 50 --batch-size -1 --save-period 10
```
使用yolov5x作为预训练模型
训练集配置保存在data/mydata.yaml
模型保存在runs/train/exp/weights/best.pt
测试
```
python3 detect.py --weights runs/train/exp/weights/best.pt --nosave --conf-thres 0 --iou-thres 0 --save-txt --source fall-test/test
```
--source为测试集路径，需要自行更改