1.找到了导致产生crop之后输出不是128*128*128的原因了。原因还是对天池数据分割的时候,对天池379分割时候得到的379tc_clean.npy结果只有半个肺右肺(以我的视角)被割掉了,而结节是在右肺当中。导致对应379tc_label.npy中的标签坐标不在379tc_clean.npy矩阵的大小范围内导致的。379tc_clean.npy替换为整肺即可。
2.运行了100遍训练的tpr在97%左右,验证的tpr在95%左右。
3.测试发现我已经将导入数据多线程数改为0了,还是显示显存已经爆掉。(测试的时候batch_size大小默认已经是1了)
4.最后将parser.add_argument('--n_test', default=8, type=int, metavar='N',# help='number of gpu for test')中default=8改为default=1后发现居然可以了。原来是这个参数控制GPU数量参与测试。难怪之前报错(只有两块GPU),同时设置了一下导入模型。
5.但是测试完之后结果很奇怪(得到的DSB2017_lbb.npy、DSB2017_pbb.npy和namelist.npy的尺寸分别是(1,4)、(98,5)和空。这个结果看得我有点蒙)只能细看测试部分的代码看哪里出了问题。