python类参数定义及数据扩展方式unsqueeze/expand
作者:徽先生 时间:2023-08-07 21:56:42
类的参数定义
将conda环境设置为ai,conda activate ai
这个文件的由来:
由于在yolov1的pytorch实现的损失函数中,看到继承了nn.Module,并且其中两个参数不像c++那里指定类型,那么他们的类型是哪里来的
这里就是在探索这样一件事
操作逻辑:
先在类中定义了构造函数以及一个自定义函数;
构造函数定义了属性S、B,自定义函数引入两个参数,对两个参数进行调用
这里就说明参数的结构是怎么样的,取决于参数被调用了什么东西,比如这里调用了
N = box1.size(0) M = box2.size(0)
说明了它是类似一个矩阵的东西,对应的box1的定义就是`torch.rand(10,4)
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
#探究属性S,B是如何产生的,以及box1、box2是如何产生的、如何调用
class yoloLoss(nn.Module):
def __init__(self,S,B):
self.S=S
self.B=B
def compute_iot(self,box1,box2):
N = box1.size(0) #调用方式就表示了变量是什么类型,这里是一个张量,其中每个元素是一个tensor,所以是N*4的张量
M = box2.size(0)
print(M,N)
yoloLoss1 =yoloLoss(10, 11)
yoloLoss1.compute_iot(torch.rand(10,4),torch.rand(11,4))
数据扩展
探究unsqueeze以及expand的使用方法,unsqueeze可以增加一个纬度,但是维度的siz只是1而已,而expand就可以将数据进行复制,将数据变为n
# 获得一开始的初始化数值:tensor([[a1,a2,a3]])
nn1=torch.rand(1,3)
print(nn1)
# unsqueeze是解压的意思,在第i个维度上进行扩展,将其扩展为tensor([[[a1,a2,a3]]])
nn1=nn1.unsqueeze(0)
print("*"*100)
print(nn1)
#利用expand对数据进行扩展
nn1=nn1.expand(1,3,3)
print("*"*100)
print(nn1)
来源:https://blog.csdn.net/weixin_42295969/article/details/126346660
标签:Python,unsqueeze,expand
0
投稿
猜你喜欢
python pandas 如何替换某列的一个值
2021-07-01 01:08:07
99%的程序员都会收藏的书单 你读过几本?
2022-10-01 19:45:59
Python特征降维知识点总结
2022-07-12 11:17:42
pytorch模型部署 pth转onnx的方法
2022-07-05 03:49:04
python 将列表中的字符串连接成一个长路径的方法
2023-04-18 14:06:25
Entity Framework使用Code First模式管理数据库
2024-01-28 04:40:43
Python3远程监控程序的实现方法
2021-07-15 15:03:50
Python join()函数原理及使用方法
2021-01-17 18:37:08
配置 SQLServer2005 以允许远程连接
2024-01-18 19:11:01
python神经网络Inception ResnetV2模型复现详解
2023-02-28 08:51:35
ASP连接MSSQL2005 数据库
2009-03-08 19:20:00
在Python中使用NLTK库实现对词干的提取的教程
2022-11-04 15:13:53
vue更多筛选项小组件使用详解
2023-07-02 17:06:01
php中防止SQL注入的最佳解决方法
2023-08-19 21:52:20
pyqt 实现在Widgets中显示图片和文字的方法
2021-12-25 07:28:16
Mysql auto_increment 重新计数(让id从1开始)
2024-01-25 17:04:54
Vue express鉴权零基础入门
2024-05-13 09:11:29
IE7下动态创建Iframe时,去除边框的办法
2009-01-19 13:56:00
获取body标签的两种方法
2024-05-08 09:39:26
Go语言学习网络编程与Http教程示例
2024-02-08 10:33:12