文件名称：pneumonia_detection：使用机器学习检测肺炎-PyTorch
文件大小：1.87MB
文件格式：ZIP
更新时间：2024-03-23 20:44:07
machine-learning computer-vision pytorch healthcare ai-health
肺炎检测 使用机器学习进行肺炎检测。 培训是在colab中完成的： 演示： 结果（混淆矩阵）： 数据 我使用了中数据集的修改版本。 我将PNEUMONIA数据集拆分为BAETERIAL PNUEMONIA和VUAL PNEUMONIA，而不是NORMAL和PNEUMONIA。 这样，数据分布更均匀，我可以区分病毒性肺炎和细菌性肺炎。 我还将验证数据集与测试数据集结合在一起，因为验证数据集每个类只有8张图像。 这是结果分布： 处理和扩充 我将图像的大小调整为150x150，并且因为某些图像已经是灰度的，所以我也将所有图像都转换为灰度。 另外，我对训练数据进行了以下转换/增强： transforms . Resize (( 150 , 150 )), transforms . Grayscale (), transforms . ToTensor (), transforms .