TransUNet：使用自己的数据集完成训练后进行测试 1.修改test.py文件，需要与train.py中设置参数保持一致 2.修改utils.py中的test_single_volume函数，可以根据自己的分割类别修改，自定义不同颜色所代表的分割种类 3.运行test.py，测试结果会以.png格式保存在前面所设置的test_save_dir中 TransUNet：记录使用自己的数据集完成训练后进行测试的学习过程 python 2024-09-09 149 点赞 2 评论 225 浏览
