
Pytorch
望舒向晚
chaos is a ladder,but for those who climb
展开
-
python修改csv某一列 读取csv移动文件
import pandas as pdpath_csv='C:\\Users\\11982\\feiyan\\train_bboxes.csv'data=pd.read_csv(path_csv)data['filename']=data['filename'].astype(str)data['filename']=data['filename'].apply(lambda x:r'.\\feiyan\\train\\'+x+'.jpg')data.to_csv(path_csv)...原创 2022-04-13 23:16:51 · 2143 阅读 · 0 评论 -
pytorch 查看模型参数 模型参数聚合
其中N是输入图像的size,F是filter的size,stride是滑动的步长。stride大于1的时候不一定能整除,这个时候,就需要在原图像上加上一层padding层,这样图像的大小就变化了卷积神经网络其实和普通的神经网络的区别在于它的输入不再是一维的向量了,而是一个三维的向量,为什么是三维的呢?这是因为图片有三个通道R,G,B。那么输出是什么呢?输出可以认为是一维的向量,比如说那图片分类举例,分为K类的话,输出就是K维的向量。使用Batch Normalization,...原创 2022-04-11 18:44:32 · 3081 阅读 · 0 评论 -
Pytorch Mask R-CNN 实例分割
Mask R-CNN通过在 Faster-RCNN 的基础上添加一个分支网络,在实现目标检测的同时,把目标像素分割出来原创 2022-03-19 00:00:00 · 1596 阅读 · 0 评论 -
PytorchGPU版本
要求Driver Version>451.67其中Driver Version是GPU 驱动版本型号,CUDA Version是CUDA 驱动 API 的型号。下载CUDA10.2检查安装是否成功:nvcc -VCUDNN v8.3.1下载需首先注册会员解压后重命名后移动到当前文件夹打开anaconda prompt,为pytorch创建一个虚拟环境,与base等区分开. 测试显卡是否可...原创 2022-03-16 00:00:00 · 1820 阅读 · 0 评论 -
创建一个2行3列的网格点矩阵
import numpy as npimport matplotlib.pyplot as plt X=np.array([[0,0.5,1],[0,0.5,1]])print("X的维度:{},shape:{}".format(X.ndim,X.shape))Y=np.array([[0,0,0],[1,1,1]])print("Y的维度:{},shape:{}".format(Y.ndim,X.shape))plt.plot(X,Y,'o--')plt.grid(True)plt....原创 2022-02-28 19:18:34 · 356 阅读 · 0 评论 -
Pytorch图像二分类代码 AlexNet
import torchimport torch.nn as nnimport torch.optim as optimimport torch.utils.dataimport torch.nn.functional as Fimport torchvisionfrom torchvision import transformsfrom PIL import Imagedef check_image(path): try: im = Image.open(pat.原创 2022-02-22 21:17:25 · 1264 阅读 · 0 评论 -
PyTorch1.7安装及基础 torchvision jupyter查看函数源码
Tensor是PyTorch的基础数据结构,是一个高维的数组,其作用类似于Numpy中的ndarray。pip list 查看python包和版本#查询文件默认存储路径import osprint(os.path.abspath('.'))原创 2022-02-20 11:00:43 · 886 阅读 · 0 评论