摘要:一、直方图 用于统计图片中各像素值: # 画一个图像各通道的直方图 def draw_hist(img): color = ('b', 'g', 'r') for i, col in enumerate(color): hist = cv.calcHist([img], [i], None, [25
阅读全文
摘要:一、继承nn.Module类并自定义层 我们要利用pytorch提供的很多便利的方法,则需要将很多自定义操作封装成nn.Module类。 首先,简单实现一个Mylinear类: from torch import nn # Mylinear继承Module class Mylinear(nn.Mod
阅读全文