CSU_Zhang的博客

原创 使用自己的数据训练MedT相关问题

首先训练前配置虚拟环境会出现无法根据yml直接生成环境的问题，首先把yml文件中第二个等号后的数字删除，并将生成环境过程中无法找到的包在yml中去掉，生成环境成功后再根据缺失的包重新进行pip下载。此部分主要检查两个方面，首先训练指令中的--gray要根据训练集是否为灰度图进行修改，其次是所有的训练集大小要相同，并和指令中的--imgsize相同。