p06358147 7c3b7c37a3 | ||
---|---|---|
data | ||
models | ||
utils | ||
LICENSE | ||
README.md | ||
cft.png | ||
global_var.py | ||
infer.py | ||
requirements.txt | ||
test.py | ||
train.py | ||
train.sh | ||
yolo2coco_l.py |
README.md
MOD。
1. 安装要求
Python 3.7+ PyTorch 1.7+ CUDA 10.1+ (推荐使用GPU进行训练和推理) 其他依赖项:请参考 requirements.txt 文件 安装依赖项:
pip install -r requirements.txt
2. 运行测试脚本
使用下面的命令进行测试:
python infer.py --weights <下载的模型路径> --data <data_sky3数据集存放路径,如'/data/dataset'>
(sky_data3数据集在dataset路径下,/data/dataset/sky_data3路径下为 train, test, val 三个文件夹和对应的json文件)
结果默认保存在detect/exp/test_pred.json文件中,其中包含检测结果的json格式数据。
3. 下载训练好的模型权重
百度云链接:https://pan.baidu.com/s/1c_NJIPKWr0nt_XPHTCUCoQ 提取码: 99im