文件名称:DenseNet-MURA-PyTorch:使用PyTorch在Standford的MURA数据集上实现DenseNet模型
文件大小:194KB
文件格式:ZIP
更新时间:2024-06-01 21:58:08
exploratory-data-analysis pytorch densenet radiology mura-dataset
使用PyTorch在MURA数据集上的DenseNet 在MURA数据集上实现169层模型的PyTorch实现,灵感来自Pranav Rajpurkar等人的论文 。 MURA是肌肉骨骼X射线照片的大型数据集,其中放射医师手动将每项研究标记为正常或异常。 重要事项: 所实现的模型是169层DenseNet,其单节点输出层使用ImageNet数据集上预先训练的模型中的权重进行初始化。 在将图像馈送到网络之前,将每个图像标准化为具有与ImageNet训练集中的图像相同的均值和标准差,并缩放为224 x 224,并通过随机的横向反转和旋转进行增强。 该模型使用了本文提到的改进的二进制交叉熵损失函数。 每次经过一段时间后,验证损失达到稳定水平,学习率就会下降10倍。 优化算法是默认参数β1= 0.9和β2= 0.999的Adam。 根据MURA数据集文件: 该模型将一个或多个用于上
【文件预览】:
DenseNet-MURA-PyTorch-master
----densenet.py(5KB)
----main.py(2KB)
----utils.py(1KB)
----train.py(4KB)
----pipeline.py(3KB)
----LICENSE(1KB)
----.gitignore(1KB)
----EDA()
--------images()
--------README.md(3KB)
--------eda.ipynb(14KB)
----README.md(3KB)