在Linux服务器上跑Python Unet程序

写于1.4日中午 在学校

我以后再也再也不拖延了。

一、遇到Out of memory的问题

解决办法:

1、把图片Resize一下。

2、把batch size改小一点。

先尝试解决办法2:

我先尝试第二种办法:batch size 原来是25,我把batch size改成2试试看。

果然可以顺利跑起来了!好开心啊~

我还是不希望resize图片,那么久吧

关于batch size和训练结果的关系,查看这一个网页:

https://www.zhihu.com/question/32673260

判断学习率设置是否良好:

其次,准确率

二、Pytorch如何保存训练好的模型

参考网址:https://blog.csdn.net/VictoriaW/article/details/72840878

我发现在代码的训练部分,有这样一段保存训练结果的代码:

这个函数的具体定义在代码下的save_checkpoint.py代码内。

意思是把state of  a net保存在checkpoint文件夹下的weights0.pth文件中。


我现在需要用的是第一种方法。

目前我已经在checkpoints文件夹下的checkpoints文件夹下保存了weights0.pth

我现在需要导入.pth文件夹。

(其实这个不需要服务器,这个在自己家里的电脑上都能跑出来。)

在test的代码里,有这样一段代码:

torch.load后面这两个参数分别表示什么意思呢?


所以我想到的解决办法是:把“weights0.pth”文件放到和result一样的文件夹下面,然后把load_weights的默认值改成“weights0.pth”,再跑。


我现在已经对原来的程序作出的改动有:

1、main里面的batchsize被我改成了从25到2

2、所有的png都被我改成了tif

3、result_visualization.py里的load_weights的默认值被我改成了“weights0.pth

4、result_visualization.py里的测试集的位置被我改变了。

三、跑test的代码

我把weights移动到对应的文件夹了。

遇到的问题1:

代码里“import matplotlib.pyplot as plt”这一句话报错。

解决办法1:

参考网站:https://stackoverflow.com/questions/49013901/importerror-cannot-import-name-path-matplotlib

就是卸载再重装

卸载重装后重启试试看。

成功解决,所以不需要看办法二了。

解决办法2:

https://blog.csdn.net/Nebula_W/article/details/79541287

这个办法可能更靠谱一点:

原因:Package中path.py的版本过高造成的。

解决方法:更改path.py的版本在7.6.1左右即可。

遇到的问题2:

代码里“import pandas as pd”这一句话报错。

参考网站:

https://stackoverflow.com/questions/30761152/how-to-solve-import-error-for-pandas

解决办法:

卸载,重装,再重启Spyder

解决!

遇到问题3:

我暂时注释掉了这两句:

遇到问题4:

类似统一原图和groundtruth图的名字的问题

用rename改一下名字就好了。

祈祷再不要报错了。

好像在跑了。

真的没有报错了。

报错是没有足够内存!

所以程序应该是对的了。

遇到问题5:

train的程序跑到第29个epoch的61%的时候忽然跑不动了。

那我就先不跑啦,先去用服务器跑test的程序。

遇到问题6:无法导入Pytorch

快速退出python模式的方法:Ctrl+z

我试过了,如果直接输入python,再输入import torch,还是不行。

解决办法1:重装Pytorch

上面显示你已经装过了pytorch

解决办法2:卸载再重装Pytorch

卸载的时候上面写:


然后我直接输入了pip install torch,居然开始安装了,之后就导入成功了。

遇到问题7:

还是CUDA OUT of memery的问题

问了学姐,学姐说解决办法是:一个图一个图地放进去跑,不要很多图一起放进去跑。

我放了一个图进去,立刻就跑好了。

遇到问题8:

跑是跑好了,但是图去哪里了呢?

我把default那里的./改成了prediction,希望能创建这么一个文件夹。

但是还是跑完了没有图。真的很奇怪。我去查源代码吧。

图就在visualization文件夹下面呀!稍微等一下就出来了。等大概1分钟。

我是在WinSCP里面看的。

四、回顾Train的代码

遇到的问题是:loss的值是负的。

我用的计算loss的方法是Dice Loss function.。

我需要先去查这个function。

Dice coefficient是常见的评价分割效果的方法之一,同样的也可以作为损失函数衡量分割的结果和标签之间的差距。

在这个class中,dice_coef函数的return是loss的结果

rna

这里有一个相关的函数定义的式子:

https://blog.csdn.net/a362682954/article/details/81226427

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,793评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,567评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,342评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,825评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,814评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,680评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,033评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,687评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,175评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,668评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,775评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,419评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,020评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,206评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,092评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,510评论 2 343

推荐阅读更多精彩内容