注意三个路径:
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的提示错误。
了解以下几点:
输入图像的读入,结果的输出,显示预测结果为中膜与管腔轮廓线,
网友评论