Skip to content

wolaituodiban/spinal_detection_baseline

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

1.本项目是Spark“数字人体”AI挑战赛——脊柱疾病智能诊断大赛的baseline
我公开了我在初赛中使用的代码,比baseline精度更高,有兴趣的同学可以移步wolaituodiban/spinal_disease_detection项目
2.本项目依赖于nn_tools,可以在github本人(wolaituodiban)项目中中自行搜索下载
3.文件结构
--code
 |--core
 |--disease 症状分类相关代码(因为是baseline,椎间盘默认预测v1,锥体默认预测v2)
 |--key_point 定位相关代码
 |--static_files 定义输出格式的静态文件
 |--structure 封装了DICOM的类,包含基础的自动寻找T2序列和中间帧的功能,拓展了一些简单空间几何变换函数
 |--data_utils.py 数据处理的常用函数
 |--dicom_utils.py 读取dicom文件的常用函数
 |--visiliation.py 简单的可视化函数
 |--main.py 入口,请在项目根目录下运行python -m code.main
--data
 |--lumbar_train150 训练集
 |--train 校验集
 |--lumbar_testA50 A榜测试集
--models 存放模型文件的目录
--predictions 存放预测结果的目录
--requirements.txt 项目的依赖,请自行安装
4.关于环境安装的建议
conda create -n py37torch15 python=3.7
conda activate py37torch15
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch
pip install -r requirements.txt
5.提交的baseline分数在0.3274左右

第一次分享baseline,如果有哪里做的不好,还请多多包含

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages