草庐IT

【39】多种方法实现矩阵的图像显示与保存

Clichong 2023-04-08 原文

如有错误,恳请指出。


一般情况下,在计算机视觉领域会出现很多次需要将图像显示或者保存到文件夹下,下面就介绍使用多种方法来实现图像的保存与显示。

文章目录

1. 从矩阵到图片的显示方法与保存

1.1 PIL实现

from PIL import Image
image= Image.fromarray(img)
image.show()  # 如果是在jupyter notebook上image就可以输出结果
image.save('image.png')

1.2 Matplotlib实现

import matplotlib.pyplot as plt
plt.imshow(image)
plt.savefig('image.png')

1.3 OpenCV实现

import cv2
cv2.imshow("image", image)
cv.imwrite('image.png', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

1.4 Numpy实现

numpy可以直接将图像矩阵保存为数据,np.array数据可以通过np.save函数直接保存在磁盘上,扩展名为.npy, 通过np.load函数直接恢复

import numpy
# 会在保存的名字后面自动加上'.npy', 所以保存的全称为'array.npy'
numpy.save('array', image)   
array = numpy.load('array.npy')
# 判断保存与读取的矩阵是否一致
(array == image).all() #返回True,代表一致

1.5 Torchvision实现

  • Torchvision实现保存图像:save_image
import torch
from torchvision.utils import save_image
import numpy

# 读取图像矩阵
array = numpy.load('array.npy')

# (h,w,c) -> (c,h,w)
image = torch.FloatTensor(array).permute(2, 0, 1).unsqueeze(0)
save_image(image, 'image.png', normalize=True)   # 保存图像

image.png:

  • Torchvision实现显示图像设置:make_grid
from torchvision import datasets, transforms
from torchvision.utils import make_grid
from torch.utils.data import DataLoader
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

# 使用matplotlib来显示图像
def show(img):
    npimg = img.numpy()
    plt.figure(figsize=(12, 12))
    plt.axis('off')   # 不显示坐标轴
    plt.imshow(np.transpose(npimg, (1,2,0)), interpolation='nearest')
    
# 加载mnist数据集
transform = transforms.Compose([
    transforms.ToTensor(),
])
dataset = datasets.MNIST(root='./dataset/',
                         train=True,
                         transform=transform,
                         download=False)
dataloader = DataLoader(dataset=dataset, batch_size=256, shuffle=True)

# 获取一批256个数据
image, label = iter(dataloader).next()

# 制作图像网格输出显示
batch_image = make_grid(image, nrow=16)
save_image(batch_image, 'image.jpg', normalize=True) # 保存图像
show(batch_image)                                    # 显示图像

image.jpg:

更多关于make_grid的参考资料:https://gist.github.com/anonymous/bf16430f7750c023141c562f3e9f2a91

ps: 除了make_gridsave_imagetorchvision.utils中还实现了几个好用的函数

  • draw_bounding_boxes:在给定图像上绘制边界框
  • draw_segmentation_masks:在给定的 RGB 图像上绘制分割蒙版
  • draw_keypoints:在给定的 RGB 图像上绘制关键点
  • flow_to_image:将光流转换为 RGB 图像

2. 从路径到图片的显示方法

2.1 PIL实现

from PIL import Image
image = Image.open('./result.jpg')
image.show()

2.2 Matplotlib实现

import matplotlib.pyplot as plt
image = plt.imread('./result.jpg')
plt.imshow(image)

2.3 OpenCV实现

import cv2
image = cv2.imread('./result.jpg')
cv2.imshow("image", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

参考资料:

1. Python中读取,显示,保存图片的方法

2. make_grid的官方测试demo

有关【39】多种方法实现矩阵的图像显示与保存的更多相关文章

  1. ruby - 如何使用 Nokogiri 的 xpath 和 at_xpath 方法 - 2

    我正在学习如何使用Nokogiri,根据这段代码我遇到了一些问题:require'rubygems'require'mechanize'post_agent=WWW::Mechanize.newpost_page=post_agent.get('http://www.vbulletin.org/forum/showthread.php?t=230708')puts"\nabsolutepathwithtbodygivesnil"putspost_page.parser.xpath('/html/body/div/div/div/div/div/table/tbody/tr/td/div

  2. ruby - 如何从 ruby​​ 中的字符串运行任意对象方法? - 2

    总的来说,我对ruby​​还比较陌生,我正在为我正在创建的对象编写一些rspec测试用例。许多测试用例都非常基础,我只是想确保正确填充和返回值。我想知道是否有办法使用循环结构来执行此操作。不必为我要测试的每个方法都设置一个assertEquals。例如:describeitem,"TestingtheItem"doit"willhaveanullvaluetostart"doitem=Item.new#HereIcoulddotheitem.name.shouldbe_nil#thenIcoulddoitem.category.shouldbe_nilendend但我想要一些方法来使用

  3. ruby - 为什么我可以在 Ruby 中使用 Object#send 访问私有(private)/ protected 方法? - 2

    类classAprivatedeffooputs:fooendpublicdefbarputs:barendprivatedefzimputs:zimendprotecteddefdibputs:dibendendA的实例a=A.new测试a.foorescueputs:faila.barrescueputs:faila.zimrescueputs:faila.dibrescueputs:faila.gazrescueputs:fail测试输出failbarfailfailfail.发送测试[:foo,:bar,:zim,:dib,:gaz].each{|m|a.send(m)resc

  4. ruby - Facter::Util::Uptime:Module 的未定义方法 get_uptime (NoMethodError) - 2

    我正在尝试设置一个puppet节点,但ruby​​gems似乎不正常。如果我通过它自己的二进制文件(/usr/lib/ruby/gems/1.8/gems/facter-1.5.8/bin/facter)在cli上运行facter,它工作正常,但如果我通过由ruby​​gems(/usr/bin/facter)安装的二进制文件,它抛出:/usr/lib/ruby/1.8/facter/uptime.rb:11:undefinedmethod`get_uptime'forFacter::Util::Uptime:Module(NoMethodError)from/usr/lib/ruby

  5. ruby-on-rails - rails : "missing partial" when calling 'render' in RSpec test - 2

    我正在尝试测试是否存在表单。我是Rails新手。我的new.html.erb_spec.rb文件的内容是:require'spec_helper'describe"messages/new.html.erb"doit"shouldrendertheform"dorender'/messages/new.html.erb'reponse.shouldhave_form_putting_to(@message)with_submit_buttonendendView本身,new.html.erb,有代码:当我运行rspec时,它失败了:1)messages/new.html.erbshou

  6. ruby-on-rails - 'compass watch' 是如何工作的/它是如何与 rails 一起使用的 - 2

    我在我的项目目录中完成了compasscreate.和compassinitrails。几个问题:我已将我的.sass文件放在public/stylesheets中。这是放置它们的正确位置吗?当我运行compasswatch时,它不会自动编译这些.sass文件。我必须手动指定文件:compasswatchpublic/stylesheets/myfile.sass等。如何让它自动运行?文件ie.css、print.css和screen.css已放在stylesheets/compiled。如何在编译后不让它们重新出现的情况下删除它们?我自己编译的.sass文件编译成compiled/t

  7. ruby-on-rails - Rails 编辑表单不显示嵌套项 - 2

    我得到了一个包含嵌套链接的表单。编辑时链接字段为空的问题。这是我的表格:Editingkategori{:action=>'update',:id=>@konkurrancer.id})do|f|%>'Trackingurl',:style=>'width:500;'%>'Editkonkurrence'%>|我的konkurrencer模型:has_one:link我的链接模型:classLink我的konkurrancer编辑操作:defedit@konkurrancer=Konkurrancer.find(params[:id])@konkurrancer.link_attrib

  8. Ruby 方法() 方法 - 2

    我想了解Ruby方法methods()是如何工作的。我尝试使用“ruby方法”在Google上搜索,但这不是我需要的。我也看过ruby​​-doc.org,但我没有找到这种方法。你能详细解释一下它是如何工作的或者给我一个链接吗?更新我用methods()方法做了实验,得到了这样的结果:'labrat'代码classFirstdeffirst_instance_mymethodenddefself.first_class_mymethodendendclassSecond使用类#returnsavailablemethodslistforclassandancestorsputsSeco

  9. ruby - 解析 RDFa、微数据等的最佳方式是什么,使用统一的模式/词汇(例如 schema.org)存储和显示信息 - 2

    我主要使用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

  10. ruby-on-rails - Rails 3.2.1 中 ActionMailer 中的未定义方法 'default_content_type=' - 2

    我在我的项目中添加了一个系统来重置用户密码并通过电子邮件将密码发送给他,以防他忘记密码。昨天它运行良好(当我实现它时)。当我今天尝试启动服务器时,出现以下错误。=>BootingWEBrick=>Rails3.2.1applicationstartingindevelopmentonhttp://0.0.0.0:3000=>Callwith-dtodetach=>Ctrl-CtoshutdownserverExiting/Users/vinayshenoy/.rvm/gems/ruby-1.9.3-p0/gems/actionmailer-3.2.1/lib/action_mailer

随机推荐