类别:背景、椎体有S、L5、L4、L3、L2、L1、T12、T11、T10、T9,椎间盘有L5/S, L4/L5, L3/L4, L2/L3, L1/L2, T12/L1, T11/T12, T10/T11, T9/T10,具体参考classes文件
参数:优化器为AdamW,学习率衰减策略为余弦退火算法,损失为交叉熵
1.train 脚本会生成训练集、验证集的loss、iou曲线、学习率衰减曲线、训练日志、数据集可视化图像等外加最后和最好的权重文件。
2.predict 脚本用于推理图像,会生成gt以及gt+image的掩膜图像
【代码做了详细
注释,可以自行下载查看,想要训练自己的数据,参考README文件,傻瓜式运行】
版权声明:
本文来源网络,所有图片文章版权属于原作者,如有侵权,联系删除。
本文网址:https://www.mushiming.com/mjsbk/13064.html