ChexNet模型的复现

1.关于论文就是吴恩达的那篇肺炎检测的论文

2.关于代码,用的是pytorch包,并在github上寻找源码参考。

记录其中的重点:

(1)关于数据的内存分配,还是用批模式训练比较合适,不然内存根本负荷不了

(2)在GPU上训练和在存CPU上训练的时候数据之间的转换问题

(3)Numpy类型的数据torch类型数据之间的转换,cpu和gpu都不同。

3.关于最后生成的热力图模式

调用的是openCV的包,利用灰度图像的伪彩图实现的。

最终的复现结果


结果不太理想,我想还有很大的可调空间。

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值