跑通分割模型过程记录

注意三个路径:

datadir(存放image、mask、models的上一级路径)、

savedir:在datadir基础上扩增的路径

imagedir:在datadir基础上扩增的路径

maskdir:在datadir基础上扩增的路径



9.9号上午十点多,运行代码后,在D:\home\insight\Desktop\data\save_models路径下保存了erfnet_ts模型 model.txt文件和opts.txt文件。

选定使用哪个模型之后,输入分类number,就可以直接获得对应的模型结构参数。

在训练模型模块,train.py中,最后对模型执行训练时,执行model = train(args,model),将参数赋值到模型进行模型的训练。

if (not os.path.exists(automated_log_path)): #dont add first line if it exists with open(automated_log_path, "a") as myfile: myfile.write("Epoch\t\tTrain-loss\t\tTest-loss\t\tTrain-IoU\t\tTest-IoU\t\tlearningRate")  在该路径下添加新文件,并写入一些内容,修改:我将not前面的括号去掉了。

2021/9/9  在自定义Dataset类时(class NeoData(Dataset):的__init__ 函数中新增变量),我自己新增了参数:mask_suffix ,因为,mask的后缀命名比image多了个‘_mask',所以就需要加入这个变量,完整的写出mask图片的路径。

DataLoader 类中的num_workers参数含义是:在加载数据时使用多少个子进程,默认设置为0,代表使用主进程进行数据加载,不同的系统环境该参数的设定会导致不同的结果。源码中的check_worker_number_rationality()函数可以对num_workers设定合理性进行检查。  linux系统可以通过len(os.sched_getaffinity(0))来直接获取,而windows则通过os.cpu()获取(并非cpu核数,而是线程数),当单process没有在指定时间内被killed,就会出现 raise RuntimeError的提示错误。

了解以下几点:

输入图像的读入,结果的输出,显示预测结果为中膜与管腔轮廓线,

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容