Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

加载断点path #12

Open
MLS2021 opened this issue Mar 6, 2023 · 4 comments
Open

加载断点path #12

MLS2021 opened this issue Mar 6, 2023 · 4 comments

Comments

@MLS2021
Copy link

MLS2021 commented Mar 6, 2023

    resnet.load_state_dict((torch.load("/root/data/others/yaoganbisai/code_6_7/models/pretrained_model/resnet50-19c8e357.pth")))

if args.model == 'deeplabv3_version_1'中,设置了path之后,会报错。错误在上面。这个resnet里面的pth是怎么生成的。

@Jiasons
Copy link

Jiasons commented Apr 22, 2023

deeplabv3_version_1是用于预测的,里面的resnet50需要自己去下载:'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth'

@hellobo2802
Copy link

deeplabv3_version_1是用于预测的,里面的resnet50需要自己去下载:'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth'

你好,你跑通了吗?

@Jiasons
Copy link

Jiasons commented May 15, 2023

deeplabv3_version_1是用于预测的,里面的resnet50需要自己去下载:'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth'

你好,你跑通了吗?

你好,这个模型跑通了,但是我预测结果全是同样的颜色。复盘可能出现问题点:
一:数据准备出问题,比如Platte中关于无效值的设定有问题。
二:预测过程中数据的读取出问题,数据的拼接也存在问题。

这个模型我修改了Platte和数据切割的很多部分,数据切割那部分写的不够完美,我自己调整了。网络的一些细节地方我也稍微调整了,但是最后仍然无法得到有效输出。

我最后是参照了bubbliiiing的代码解决的https://github.com/bubbliiiing/deeplabv3-plus-pytorch。可以利用这个项目中data process中的切割代码对训练集和测试集切割,将训练集准备好后放入bubbling的代码之中可以顺利跑通(唯一需要注意无效值设定的代码)。之后只需要对测试集进行预测,再手写一个代码将图片拼接即可

@zhouchuanxu
Copy link

deeplabv3_version_1是用于预测的,里面的resnet50需要自己去下载:'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth'

你好,你跑通了吗?

你好,这个模型跑通了,但是我预测结果全是同样的颜色。复盘可能出现问题点: 一:数据准备出问题,比如Platte中关于无效值的设定有问题。 二:预测过程中数据的读取出问题,数据的拼接也存在问题。

这个模型我修改了Platte和数据切割的很多部分,数据切割那部分写的不够完美,我自己调整了。网络的一些细节地方我也稍微调整了,但是最后仍然无法得到有效输出。

我最后是参照了bubbliiiing的代码解决的https://github.com/bubbliiiing/deeplabv3-plus-pytorch。可以利用这个项目中data process中的切割代码对训练集和测试集切割,将训练集准备好后放入bubbling的代码之中可以顺利跑通(唯一需要注意无效值设定的代码)。之后只需要对测试集进行预测,再手写一个代码将图片拼接即可

你好xd,可以分享一下你更新之后的代码吗

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants