01 2020 档案

[图像处理]基于 PyTorch 的高斯核卷积
摘要：``` import torch import numpy as np import torch.nn as nn import torch.nn.functional as F import cv2 import matplotlib.pyplot as plt from PIL import Image class GaussianBlurConv(nn.Module): def __init 阅读全文
posted @ 2020-01-17 22:23 虔诚的树 阅读(3128) 评论(0) 推荐(0)
[PyTorch] torch.squeee 和 torch.unsqueeze()
摘要：torch.squeeze 分为两种情况: 不指定维度 或 指定维度 1. 不指定维度 Example 2. 指定维度 & `output: (A, 1, B)` & `output: (A, B)` Example Note: The returned tensor shares the stor 阅读全文
posted @ 2020-01-17 13:50 虔诚的树 阅读(328) 评论(0) 推荐(0)

点击右上角即可分享
微信分享提示