logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

【python实现图像矫正 全能扫描王!】

全能扫描王app里面有一个对图像进行矫正的功能,实际上是通过四点法的仿射变换来进行操作的。选取文档的四个角点,然后计算仿射变换矩阵来对图像进行矫正。下面通过一段python程序来进行简单实现,功能为:通过顺时针点击文档的左上,右上,右下,左下四个角点来对其进行图像矫正。代码如下:# coding=utf-8# 导入相应的python包import numpy as npimport c...

#python
【matplotlib 叠加图片显示热力图 attention map】

有的时候需要看attention map集中在图片当中的哪些区域,或者需要看重构误差在哪些区域比较大,所以需要将“热力图”和图片叠加显示,先看效果图:代码如下:import matplotlib.pyplot as pltimg_path = 'IMAGE_PATH'mask_path = 'MASK_PATH...

#matplotlib
pytorch做inference的时候如何减少显存占用

1、采用with torch.no_grad():with torch.no_grad():tc_data = netG_A(input_data, d_f)相比于直接做inference,加上with torch.no_grad()能够占用更少的显存。即使设置了netG_A.eval(),占用显存和没设置时是一样的。难道eval()模式也还会保留中间梯度?tc_data = netG_A(inp

#python#pytorch
【pytorch: can't optimize a non-leaf Tensor】

需要先把Tensor放入到GPU中,然后再设置Tensor.requires_grad=True。from torch import optimweights = torch.rand(2,1,128,416)weights.requires_grad = Trueweights = weights.cuda()optimizer = optim.Adam([weights], l...

#pytorch
./ 运行bash脚本文件出现 报错信息 /usr/bin/env: “bash\r“: 没有那个文件或目录

用 ./ 运行bash脚本文件出现 报错信息 /usr/bin/env: "bash\r": 没有那个文件或目录错误原因:这主要是因为bash后面多了\r这个字符的原因。在linux终端下,输出\r会什么都不显示,只是把光标移到行首。于是终端虽然输出了/usr/bin/env bash,但是碰到\r后,光标会被移到行首,接着输出了:No such file or directory把前面的覆盖掉了

#python#linux#bash +1
[ubuntu下安装tensorflow和keras]

keras 是需要tensorflow 支持的,所以在安装keras之前需要安装tensorflow。1、安装tensorflow具体步骤可以参考tensorflow中文社区,点击打开链接执行:pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-lin...

#keras#tensorflow
查看gpu使用情况并且查看是那个用户在使用gpu【gpustat】

最近看了师兄的显示屏,get到了两个好棒的功能,一个是linux的shell分屏,一个是实时显示gpu的状态和使用情况。分屏指令:tmux这个指令有好多介绍了,就不详细赘述,https://www.cnblogs.com/kevingrace/p/6496899.htmlgpu使用情况:gpustathttps://github.com/wookayin/gpustat详细...

到底了