Python GPU排队脚本
最近任务的模态太多了,要跑很多次训练脚本
于是就写了这个脚本,主要功能就是检测GPU的剩余显存,如果空闲就跑列表里没启动的任务,任务这边用命令行调用的方式,还是比较灵活的
继续阅读“Python GPU排队脚本”Counterfactual Attention Learning (CAL)原理及在3DViT上的实现
Grad-CAM对3DVIT结果进行可视化的原理与pytorch实现
SimpleITK无法处理中文路径的解决方法
通过改变工作路径的方式规避输入中文路径,如果需要中文文件名也是同理,先输入保存后用shutils.move改名即可。
# 改变当前工作目录为输出文件夹所在目录
if not os.path.exists(output_path):
os.makedirs(output_path)
os.chdir(output_path)
# 将图像转换为 NIfTI 格式并保存为 .nii.gz
output_file = 'tissues_ori_src.nii.gz'
sitk.WriteImage(image, output_file)