- 博客(14)
- 收藏
- 关注
RSS订阅原创 调用gensim中的word2vec
"""功能：测试gensim使用，处理中文语料"""from gensim.models import word2vecimport loggingfrom gensim.models import Word2Vecfrom seaborn import heatmapfrom matplotlib import pyplotimport numpy as np# 主程序logging.basicConfig(format='%(asctime)s : %(levelname)s..
2022-01-25 20:26:47
1418
原创 自己实现word2vec
import numpy as npimport torchimport torch.nn as nnimport torch.optim as optimfrom torch.autograd import Variableimport matplotlib.pyplot as pltfrom gensim.models import word2vecdtype = torch.FloatTensor# 3 Words Sentence# 分词，词汇表构造以及词汇索引的构造sent.
2022-01-25 20:24:41
257
原创 VGG16_GPU
import torchimport torch.nn as nnimport torchvision.models as modelsimport matplotlib.pyplot as pltimport torchvision.datasetsimport torchvision.transforms as transformsfrom collections.abc import Iterable# 检测是否可以使用GPUfrom torch.utils.data import .
2022-01-24 19:56:37
444
原创 VGG16_cpu
import torchimport torch.nn as nnimport torchvision.models as modelsimport matplotlib.pyplot as pltimport torchvision.datasetsimport torchvision.transforms as transformsfrom collections.abc import Iterable# 检测是否可以使用GPUfrom torch.utils.data import .
2022-01-24 19:56:02
757
原创 自己实现resnet18网络结构
import torchimport torch.nn as nn# 定义一个Residual模块class Residual(nn.Module): def __init__(self,in_channels,out_channels,stride=1): super(Residual, self).__init__() self.stride = stride self.conv1 = nn.Conv2d(in_channels,out..
2022-01-24 19:54:57
874
原创 GPU实现
import torchimport torchvision.datasetsfrom matplotlib import pyplot as pltfrom torch.utils.data import DataLoaderfrom torchvision.transforms import transformsimport torch.nn as nnimport osos.environ['KMP_DUPLICATE_LIB_OK'] = 'True'# 调用torchvis.
2022-01-24 19:53:42
139
原创 调用TorchVision中训练好的ResNet网络训练CIFAR10
import torchimport torchvision.datasetsfrom matplotlib import pyplot as pltfrom torch.utils.data import DataLoaderfrom torchvision.transforms import transformsimport torch.nn as nnimport imutils# 调用torchvision中的models中的resnet网络结构import torchvisi.
2022-01-24 19:52:27
3695
1
原创 CIFAR10_ResNet
import torchimport torchvision.datasetsfrom matplotlib import pyplot as pltfrom torch.utils.data import DataLoaderfrom torchvision.transforms import transformsimport torch.nn as nnimport osos.environ['KMP_DUPLICATE_LIB_OK'] = 'True'# 调用torchvis.
2022-01-24 19:50:38
783
1
原创 ResNet
# 调整已经训练好的ResNet网络class ResNet(nn.Module): def __init__(self, num_classes=10): # num_classes，此处为 二分类值为2 super(ResNet, self).__init__() net = models.resnet18(pretrained=True) # 从预训练模型加载resnet18网络参数 net.classifier = nn.Sequen.
2022-01-24 19:49:41
2436
原创 CIFAR10
import torch.optimfrom torch.utils.tensorboard import SummaryWriter# from model import * # 注意model文件和train文件是在同一个目录底下import torchvision.datasetsfrom torch import nnimport torchvision.transforms as transformsfrom torch.nn import *from torch.utils.
2022-01-24 19:46:04
361
原创 不调用torch.nn中函数，自己实现cnn的功能进行mnist手写体识别
import torchvision.transformsfrom torch.utils.data import DataLoaderfrom torch.utils.tensorboard import SummaryWriterfrom torchvision import datasetsimport numpy as npimport matplotlib.pyplot as plt# 1.导入mnist数据集train_dataset = datasets.MNIST(root.
2022-01-24 19:44:12
541
1
原创 CNN实现mnist手写识别/CIFAR10数据集（PyTorch）
import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as pltimport osimport cv2from torchvision import transformstorch.manual_seed(1) # 使用随机化种子使神经网络的初始化每次都相同# 超参数EPOCH = 10 # 训练整批数据的次数BAT.
2022-01-24 19:39:24
853
原创 OpenCV
图像的的基本操作去除图片噪音的几种方式如下：均值滤波方框滤波高斯滤波中值滤波(效果最好)candy进行边缘检测(其中用的是sobel算子)import cv2 as cvimport numpy as npimage = cv.imread("D:\APP\PycharmProjects\pythonProject\learn_opencv\images\sp_noise.png")# cv.imshow("image",image)# 均值滤波# 简单的平均卷
2021-11-25 19:20:34
1228
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人