Go to file
p06358147 7c3b7c37a3 Update README.md 2024-11-23 22:16:13 +08:00
data test 2024-11-22 23:53:31 +08:00
models test 2024-11-22 23:53:31 +08:00
utils test 2024-11-22 23:53:31 +08:00
LICENSE test 2024-11-22 23:53:31 +08:00
README.md Update README.md 2024-11-23 22:16:13 +08:00
cft.png test 2024-11-22 23:53:31 +08:00
global_var.py test 2024-11-22 23:53:31 +08:00
infer.py test 2024-11-22 23:53:31 +08:00
requirements.txt test 2024-11-22 23:53:31 +08:00
test.py test 2024-11-22 23:53:31 +08:00
train.py test 2024-11-22 23:53:31 +08:00
train.sh test 2024-11-22 23:53:31 +08:00
yolo2coco_l.py test 2024-11-22 23:53:31 +08:00

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