structure of project
--project
main.py
--data
--train
--val
to download data and trained weight link:https://pan.baidu.com/s/10E0CIehRMUA7zepmSbtjjA code:wabv
python main.py train
load the last saved weight
python main.py test --ckp=weights_19.pth
项目文件分布如下
--project
main.py
--data
--train
--val
数据和权重可以使用百度云下载 链接:https://pan.baidu.com/s/10E0CIehRMUA7zepmSbtjjA 提取码:wabv
python main.py train
加载权重,默认保存最后一个权重
python main.py test --ckp=weights_19.pth