美文网首页
跑通分割模型过程记录

跑通分割模型过程记录

作者: Allard_c205 | 来源:发表于2021-09-13 16:52 被阅读0次

    注意三个路径:

    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的提示错误。

    了解以下几点:

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

    相关文章

      网友评论

          本文标题:跑通分割模型过程记录

          本文链接:https://www.haomeiwen.com/subject/ukwiwltx.html