基于改进的密集连接网络的X线胸片分类装置

摘要

本发明涉及一种基于改进的密集连接网络的X线胸片分类装置，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时用于实现下面所述的方法步骤：第一步，胸片图预处理；第二步，构建和训练深度卷积神经网络：先对分类网络DenseNet121进行修改，在该网络中以密集连接的方式加入挤压—激励模块，形成一个改进的密集连接网络，再构建符合本任务的损失函数，并利用预处理后的胸片图对网络进行训练；第三步，测试网络并选出最佳的网络模型：将每一轮训练得到的网络模型在测试集上进行测试，并选出平均AUC值最高的网络模型作为最终的模型。