Redian新闻
>
Kaggle医学影像新赛事:腹部创伤检测

Kaggle医学影像新赛事:腹部创伤检测

公众号新闻


最近Kaggle发布了RSNA 2023 Abdominal Trauma Detection 腹部创伤检测竞赛。本次竞赛是一个计算机视觉任务


参赛者需要在创伤患者的CT扫描中识别出损伤类型及程度。本次比赛将评估提交数据中的每种伤害类型的样本加权对数损失的平均值,及该指标生成的any iniury伤害预测值。


为了帮助同学们冲分拿牌,我联合Kaggle前1000大神Mozak老师,带来赛题讲座,详解高分baseline。课程价值198元,限时免费观看扫码即刻解锁!


扫码添加课程顾问

赛题讲座免费看,详解高分baseline!


本次比赛的思路为:


数据预处理和图像特征提取

使用Python库如pydicom来加载这些图像。由于DICOM数据可能包含多个图像序列,可以根据patient_idseries_id将它们组织起来。

模型选择

可以选择使用经典的卷积神经网络架构,如ResNet、DenseNet或EfficientNet。

模型训练和验证

将数据分为训练集和验证集,用于训练和调整模型。

模型推断和测试集评估

在训练好的模型上进行推断,对测试集中的图像进行预测。


扫码添加课程顾问

赛题讲座免费看,详解高分baseline

以下为部分关键代码:


 将DICOM转换为jpg

def dicom_to_image(dicom_image):    """    Read the dicom file and preprocess appropriately.    """    pixel_array = dicom_image.pixel_array        if dicom_image.PixelRepresentation == 1:        bit_shift = dicom_image.BitsAllocated - dicom_image.BitsStored        dtype = pixel_array.dtype         new_array = (pixel_array << bit_shift).astype(dtype) >>  bit_shift        pixel_array = pydicom.pixel_data_handlers.util.apply_modality_lut(new_array, dicom_image)        if dicom_image.PhotometricInterpretation == "MONOCHROME1":        pixel_array = 1 - pixel_array        # transform to hounsfield units    intercept = dicom_image.RescaleIntercept    slope = dicom_image.RescaleSlope    pixel_array = pixel_array * slope + intercept        # windowing    window_center = int(dicom_image.WindowCenter)    window_width = int(dicom_image.WindowWidth)    img_min = window_center - window_width // 2    img_max = window_center + window_width // 2    pixel_array = pixel_array.copy()    pixel_array[pixel_array < img_min] = img_min    pixel_array[pixel_array > img_max] = img_max        # normalization    pixel_array = (pixel_array - pixel_array.min())/(pixel_array.max() - pixel_array.min())        return (pixel_array * 255).astype(np.uint8)


 加载基础模型

model = torchvision.models.resnet34(True)model.fc = torch.nn.Linear(512, 14)model.conv1 = nn.Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
model = model.cuda()pos_weight = torch.Tensor([1,2,1,6,1,2,4,1,2,4,1,2,4,6]).cuda()criterion = nn.BCEWithLogitsLoss(pos_weight=pos_weight)optimizer = torch.optim.SGD(model.parameters(), 0.0001)


 模型预测

pred_list = []for pid in train_pids[:10]:        pid_paths_dcm_paths = glob.glob('/kaggle/input/rsna-2023-abdominal-trauma-detection/train_images/' + str(pid) + '/*/*')    pid_paths_dcm_paths.sort()        pid_paths_dcm_paths = pid_paths_dcm_paths[-5:]    imgs = [Image.fromarray(dicom_to_image(pydicom.read_file(x))) for x in pid_paths_dcm_paths]    imgs = [transform(x) for x in imgs]        imgs = torch.cat(imgs, 0)    imgs = imgs[:, None, :, :]    with torch.no_grad():        imgs = imgs.cuda()        output = model(imgs)[:, :-1]        pred = torch.sigmoid(output).data.cpu().numpy().round(3)        pred = pred.mean(0)            pred_list.append(pred)


扫码添加课程顾问

赛题讲座免费看,详解高分baseline


微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
现在怎么穿纸尿裤的都这么厉害了?不像新学的像没忘干净校招 | GE医疗2024秋季校园招聘正式启动!百年医药外企,世界五百强,多地有岗,一对一带教专访影动医疗创始人龚南杰:打造全球领先的AI医学影像技术Rescue Teams Struggle to Reach Isolated Areas in Flood-Hit Hebei10位投资人,100家独角兽,为大企业高管写出内部创新的方法论观点|宋伟:BRICS play bigger role in improving global governance「影像新生力」南京苏州站活动回顾,更多城市持续报名中!【北京】名企参访丨探索GE医疗服务转型创新之路Major Setbacks: Students Struggle with Scrapped Majors太湖美!太湖臭?博士申请 | 佛罗里达大学邵尉老师招收深度学习/CV/医学影像方向全奖博士生百万奖金池!目标检测、分割、测距、融合、规划、控制等新赛事全面启动!2023中国智能网联汽车算法挑战赛(CIAC 2023)来了不必等黑五,lululemon现在就5折了!多款leggings史低价入手父女情谊的复盘:聊聊电影“Aftersun“「影像新生力」长沙站回顾 解锁设备新体验【内含baseline】Kaggle机器学习新赛指南!文末送书~KDD 2023 | GPT时代医学AI新赛道:16万张图片、70万问答对的临床问答数据集MIMIC-Diff-VQA发布剧院新闻|Dear England将转西区,本周末纪录片影像饕餮继续(全)2023年可授予证书的赛事名录,共2279项赛事!人工智能:阶级斗争的归来?| AI: The Return Of Marxist Class Struggle?当吴妈第一次内部创业《月光爱人》&《想你的365天》公开出售GitHub star、Kaggle点赞,「黑市」越来越明目张胆了吗?赛事调整 | 体育总局办公厅关于调整2023年可授予运动员技术等级称号部分赛事名录的通知第五章 列宁主义救中国 (2)KDD2023 | GPT时代医学AI新赛道:16万张图片、70万问答对的临床问答数据集MIMIC-Diff-VQA发布UCCA影像艺术交流丨“马蒂斯的马蒂斯”展览系列公共影像:《文明》系列第1-2集108万奖金!目标检测、图像分类等赛事!2023长三角(芜湖)人工智能视觉算法大赛全面启动!目标检测新赛事!"达观杯"智能文档版面分析大赛来了!丰厚奖金+面试优先录用!老妈的大爱!2023THE世界大学影响力排名!“出海四小龙”新战事:SHEIN 上半年赚了更多的钱,TikTok要去美国做电商,Temu上线韩国站点丨Going Global阿里妈妈往事,当吴妈第一次内部创业GE医疗2024秋季校园招聘正式启动!百年医药外企,世界五百强,多地有岗,一对一带教「影像新生力」再度启航!7城巡讲现正式开启报名
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。