P5103081-植物病害识别练习项目

0.项目来源

本项目Github地址为:

GitHub - spytensor/plants_disease_detection: Ai Challenger 2018 Competitions 农作物病害检测

1.必要准备

数据集 百度网盘 请输入提取码 提取码：iksk

下载项目：git clone https://github.com/spytensor/plants_disease_detection.git

2.环境配置

2.1硬件环境配置

显卡：Nvidia Geforce RTX5090 @ 32GB * 1

2.2软件环境配置

Linux：Ubuntu 24.04LTS（WSL2）

Anaconda：最新版本

CUDA：13.0

Python version info: 3.10.19 (main, Oct 21 2025, 16:43:05) [GCC 11.2.0]
PyTorch version info: 2.10.0.dev20251026+cu130

2.3环境配置详细过程

第一步：将测试集图片复制到 data/test/ 下

本步骤注意事项：将图片复制到test文件夹下即可（打开test文件夹应当显示一大堆图片）。如果把images文件夹复制过来了，需要在根目录下的config.py中修改 test_data="./data/test/images/"

第二步：将训练集合验证集中的图片都复制到 data/temp/images/ 下，将两个 json 文件放到 data/temp/labels/ 下

（训练集和验证集图片位置图参考上面）

执行 move.py 文件

python move.py

执行 main.py 进行训练

python main.py

3.模型效果

|------------ VALID -------------|----------- TRAIN -------------|------Accuracy------|------------|

lr iter epoch | loss top-1 top-2 | loss top-1 top-2 | Current Best | time |

0.0000 39.0 39.0 | 0.313 87.682 99.209 | 0.279 88.877 99.128 | 87.957344 | 5 hr 31 min