1.关于论文就是吴恩达的那篇肺炎检测的论文
2.关于代码,用的是pytorch包,并在github上寻找源码参考。
记录其中的重点:
(1)关于数据的内存分配,还是用批模式训练比较合适,不然内存根本负荷不了
(2)在GPU上训练和在存CPU上训练的时候数据之间的转换问题
(3)Numpy类型的数据torch类型数据之间的转换,cpu和gpu都不同。
3.关于最后生成的热力图模式
调用的是openCV的包,利用灰度图像的伪彩图实现的。
最终的复现结果
结果不太理想,我想还有很大的可调空间。