Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
|
|
8 months ago | |
|---|---|---|
| .gitignore | 10 months ago | |
| LICENSE | 10 months ago | |
| README.md | 8 months ago | |
| class_training dataset.png | 8 months ago | |
| data_analysis.py | 8 months ago | |
| main.py | 8 months ago | |
| metrics_curve.png | 8 months ago | |
| requirements.txt | 8 months ago | |
================================
项目介绍
本项目基于 Jittor 深度学习框架,实现了对乳腺癌超声影像的精准分类。main.py 文件集成
了多种先进的深度学习技术来解决数据类别不平衡问题,包括 Focal Loss、类别权重、数据
增强等,使用 ResNet34 作为基础网络架构,模型轻量化的同时保持了不错的性能,训练以
及推理速度都较快。
主要特性:
├── main.py # ResNet34 主训练文件(集成所有改进)
├──data_analysis.py # 数据集具体信息可视化文件
├── requirements.txt # 项目依赖
├── TrainSet/ # 训练数据集
│ ├── images/train/ # 训练图像
│ └── labels/ # 标签文件
│ ├── train.txt # 训练集标签
│ └── val.txt # 验证集标签
├── TestSetA/ # 测试数据集
└── model_save/ # 模型保存目录
核心功能模块:
安装 Python 3.7+
安装项目依赖:
pip install -r requirements.txt
依赖包说明:
训练 ResNet34 模型:
python main.py
使用训练好的模型进行推理:
python main.py --testonly --dataroot ./TestSetA --loadfrom ./model_save/res34-best6-
19.pkl --result_path ./result.txt
参数说明: