目录
之前已经将unet的网络模块、dataset数据加载和train训练数据已经解决了，这次要将unet网络去分割图像，下面是之前的链接
unet 网络：UNet - unet网络
dataset 数据处理：UNet - 数据加载 Dataset
train 网络训练：UNet - 训练数据train
待分割的图像如下：
存放的路径在U-net项目的predict里面
我们的目标是将predict里面所有的图片分割出来，按照名称顺序保存在result文件夹里面：
首先定义图片的预处理，按照dataset里面相同的方式进行预处理
然后是加载网络的模型和网络参数
然后加载predict里面所有待处理图片的路径
需要注意的是，os.listdir 加载的只是里面每个图片，并不是图片的具体路径。tests_path 里面的内容如下面的注释所示：
接下来就可以分割图片了
因为tests_path 里面每个文件是 x.png 即文件名+后缀的方式。通过split的 '.' 分割成x和后缀名png的形式，[-2]代表取倒数第二个值，就可以将每个文件名x取出来，然后将路径拼接就可以存放到result里面
open图像的时候，也要注意，test_path 只是遍历tests_path 里面的文件，需要加上之前的predict路径才能正确的读取到每个待分割的图片
因为这里处理图像会改变size成480*480的形式，想要将输出的结果保持不变的话，在网络预测前将图像的大小保存下来就可以了。（注：这里的size和opencv里面的shape返回值是反过来的）
这里不清楚的可以通过调试，打印每个变量的内容看一下就可以了
接下来就是网络预测的部分，这里输出的size是（batch，channel，height，width），因为这里的batch是1，channel 灰度图片因此也是1，这里通过squeeze将1的维度删去，只需要图像的大小
下面是squeeze的用法
然后图像保存的话，要转到cpu上面 ，这一步不知道为啥，但是不加这一步会报错
最后就是保存图像了，将网络的结果二值化后，还原图像再保存就可以了
predict里面待预测的图片
result 里面分割好的图片
下面是 参考文章 博主的分割结果
对比发现，有些小的细节会丢失，但是大概的轮廓分割出来了
完整的项目可以在 这里 下载
import numpy as np
import torch
import cv2
from model import UNet
from torchvision import transforms
from PIL import Image
import os
# 预处理
transform = transforms.Compose([
transforms.Resize((480,480)), # 缩放图像
transforms.ToTensor(),
])
# 加载模型
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
net = UNet(in_channels=1, num_classes=1)
net.load_state_dict(torch.load('Unet.pth', map_location=device))
net.to(device)
# 测试模式
net.eval()
# 读取所有图片路径
tests_path = os.listdir('./predict/') # 获取 './predict/' 路径下所有文件,这里的路径只是里面文件的路径
''''
print(tests_path)
['0.png', '1.png', '10.png', '11.png', '12.png', '13.png', '14.png',
'15.png', '16.png', '17.png', '18.png', '19.png', '2.png', '20.png',
'21.png', '22.png', '23.png', '24.png', '25.png', '26.png', '27.png',
'28.png', '29.png', '3.png', '4.png', '5.png', '6.png', '7.png', '8.png', '9.png']
'''
with torch.no_grad(): # 预测的时候不需要计算梯度
for test_path in tests_path: # 遍历每个predict的文件
save_pre_path = './result/'+test_path.split('.')[-2] + '_res.png' # 将保存的路径按照原图像的后缀，按照数字排序保存
img = Image.open('./predict/' +test_path) # 预测图片的路径
width,height = img.size[0],img.size[1] # 保存图像的大小
img = transform(img)
img = torch.unsqueeze(img,dim = 0) # 扩展图像的维度
pred = net(img.to(device)) # 网络预测
pred = torch.squeeze(pred) # 将(batch、channel)维度去掉
pred = np.array(pred.data.cpu()) # 保存图片需要转为cpu处理
pred[pred >= 0] = 255 # 处理结果二值化
pred[pred < 0] = 0
pred = np.uint8(pred) # 转为图片的形式
pred = cv2.resize(pred,(width,height),cv2.INTER_CUBIC) # 还原图像的size
cv2.imwrite(save_pre_path, pred) # 保存图片
Rails2.3可以选择随时使用RouteSet#add_configuration_file添加更多路由。是否可以在Rails3项目中做同样的事情？ 最佳答案 在config/application.rb中:config.paths.config.routes在Rails3.2(也可能是Rails3.1)中，使用:config.paths["config/routes"] 关于ruby-on-rails-Rails3中的多个路由文件，我们在StackOverflow上找到一个类似的问题
我有多个ActiveRecord子类Item的实例数组，我需要根据最早的事件循环打印。在这种情况下，我需要打印付款和维护日期，如下所示:ItemAmaintenancerequiredin5daysItemBpaymentrequiredin6daysItemApaymentrequiredin7daysItemBmaintenancerequiredin8days我目前有两个查询，用于查找maintenance和payment项目(非排他性查询)，并输出如下内容:paymentrequiredin...maintenancerequiredin...有什么方法可以改善上述(丑陋的)代
我主要使用Ruby来执行此操作，但到目前为止我的攻击计划如下:使用gemsrdf、rdf-rdfa和rdf-microdata或mida来解析给定任何URI的数据。我认为最好映射到像schema.org这样的统一模式，例如使用这个yaml文件，它试图描述数据词汇表和opengraph到schema.org之间的转换:#SchemaXtoschema.orgconversion#data-vocabularyDV:name:namestreet-address:streetAddressregion:addressRegionlocality:addressLocalityphoto:i
我需要从一个View访问多个模型。以前，我的links_controller仅用于提供以不同方式排序的链接资源。现在我想包括一个部分(我假设)显示按分数排序的顶级用户(@users=User.all.sort_by(&:score))我知道我可以将此代码插入每个链接操作并从View访问它，但这似乎不是“ruby方式”，我将需要在不久的将来访问更多模型。这可能会变得很脏，是否有针对这种情况的任何技术？注意事项:我认为我的应用程序正朝着单一格式和动态页面内容的方向发展，本质上是一个典型的网络应用程序。我知道before_filter但考虑到我希望应用程序进入的方向，这似乎很麻烦。最终从任何
我有一个具有一些属性的模型:attr1、attr2和attr3。我需要在不执行回调和验证的情况下更新此属性。我找到了update_column方法，但我想同时更新三个属性。我需要这样的东西:update_columns({attr1:val1,attr2:val2,attr3:val3})代替update_column(attr1,val1)update_column(attr2,val2)update_column(attr3,val3) 最佳答案 您可以使用update_columns(attr1:val1,attr2:val2
我正在尝试修改当前依赖于定义为activeresource的gem:s.add_dependency"activeresource","~>3.0"为了让gem与Rails4一起工作，我需要扩展依赖关系以与activeresource的版本3或4一起工作。我不想简单地添加以下内容，因为它可能会在以后引起问题:s.add_dependency"activeresource",">=3.0"有没有办法指定可接受版本的列表？~>3.0还是~>4.0？ 最佳答案 根据thedocumentation,如果你想要3到4之间的所有版本，你可以这
有时我需要处理键/值数据。我不喜欢使用数组，因为它们在大小上没有限制(很容易不小心添加超过2个项目，而且您最终需要稍后验证大小)。此外，0和1的索引变成了魔数(MagicNumber)，并且在传达含义方面做得很差(“当我说0时，我的意思是head...”)。散列也不合适，因为可能会不小心添加额外的条目。我写了下面的类来解决这个问题:classPairattr_accessor:head,:taildefinitialize(h,t)@head,@tail=h,tendend它工作得很好并且解决了问题，但我很想知道:Ruby标准库是否已经带有这样一个类？ 最佳
我正在尝试按0-9和a-z的顺序创建数字和字母列表。我有一组值value_array=['0','1','2','3','4','5','6','7','8','9','a','b'，'光盘'，'e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','','u','v','w','x','y','z']和一个组合列表的数组，按顺序，这些数字可以产生x个字符，比方说三个list_array=[]和一个当前字母和数字组合的数组(在将它插入列表数组之前我会把它变成一个字符串，]current_combo['0','0','0']
是否有可能:before_filter:authenticate_user!||:authenticate_admin! 最佳答案 before_filter:do_authenticationdefdo_authenticationauthenticate_user!||authenticate_admin!end 关于ruby-on-rails-before_filter运行多个方法，我们在StackOverflow上找到一个类似的问题： https://
我正在使用Rails3.1并在一个论坛上工作。我有一个名为Topic的模型，每个模型都有许多Post。当用户创建新主题时，他们也应该创建第一个Post。但是，我不确定如何以相同的形式执行此操作。这是我的代码:classTopic:destroyaccepts_nested_attributes_for:postsvalidates_presence_of:titleendclassPost...但这似乎不起作用。有什么想法吗？谢谢! 最佳答案 @Pablo的回答似乎有你需要的一切。但更具体地说...首先改变你View中的这一行对此#