pneumonia_detection:使用机器学习检测肺炎 PyTorch 源码

肺炎检测 使用机器学习进行肺炎检测。 培训是在colab中完成的: 演示: 结果(混淆矩阵): 数据 我使用了中数据集的修改版本。 我将PNEUMONIA数据集拆分为BAETERIAL PNUEMONIA和VUAL PNEUMONIA,而不是NORMAL和PNEUMONIA。 这样,数据分布更均匀,我可以区分病毒性肺炎和细菌性肺炎。 我还将验证数据集与测试数据集结合在一起,因为验证数据集每个类只有8张图像。 这是结果分布: 处理和扩充 我将图像的大小调整为150x150,并且因为某些图像已经是灰度的,所以我也将所有图像都转换为灰度。 另外,我对训练数据进行了以下转换/增强: transforms . Resize (( 150 , 150 )), transforms . Grayscale (), transforms . ToTensor (), transforms .