记录小代码
日常一些自己用到的一些代码,记录一下,方便使用
CV小Rookie
love life~
展开
-
PyTorch 模型保存,断点训练
PyTorch 模型保存,断点训练原创 2022-09-21 09:34:10 · 1632 阅读 · 0 评论 -
Pytorch 中打印网络结构及其参数的方法与实现
pytorch 中打印网络结构及其参数的方法与实现原创 2022-09-17 12:22:46 · 8935 阅读 · 2 评论 -
图像线性融合
import cv2for i in range(1,5001): img1 = cv2.imread('/Users/like/Desktop/mix/1/'+str(i)+'.jpg') img2 = cv2.imread('/Users/like/Desktop/mix/2/'+str(i)+'.jpg') #img2 = cv2.resize(img2,(512,512))#统一图片大小 dst = cv2.addWeighted(img1,0.5,img2,.原创 2022-05-09 10:45:17 · 311 阅读 · 0 评论 -
图像形变(插值方法)
类似于iphone相册的水平矫正双线性插值from PIL import Imageimport matplotlib.pyplot as pltimport numpy as npimport mathdef BiLinear_interpolation(img,dstH,dstW): scrH,scrW,_=img.shape print(img) img=np.pad(img,((0,1),(0,1),(0,0)),'constant') print(原创 2022-05-06 19:01:40 · 304 阅读 · 0 评论 -
JPEG2jpg
import osimport stringdir1 = "11/"dir2 = "test/"# 读取函数,用来读取文件夹中的所有函数,输入参数是文件名def read_directory(directory_name): i=0 for filename in os.listdir(directory_name): print(filename) if "DS_Store" in filename: continue.原创 2022-05-06 18:43:16 · 99 阅读 · 0 评论 -
图像resize
import cv2import os# 读取函数,用来读取文件夹中的所有函数,输入参数是文件名def read_directory(directory_name): for filename in os.listdir(directory_name): print(filename) if "DS_Store" in filename: continue img = cv2.imread(directory_name.原创 2022-05-06 18:39:50 · 227 阅读 · 0 评论 -
图像合并水平拼接
import PIL.Image as Imageimport os# from IPython import embed# embed()# 定义图像拼接函数def image_compose(imag, imag_1): src = os.path.join(os.path.abspath(IMAGE_SAVE_PATH), img) to_image = Image.new('RGB', (2 * h, 1 * w)) # 创建一个新图 # 把两张图片按顺序粘贴.原创 2022-05-06 18:38:53 · 235 阅读 · 0 评论