【관련 추천: Python3 동영상 튜토리얼】
conda 환경을 ai로 설정, conda activate ai
conda activate ai
这个文件的由来:
由于在yolov1的pytorch实现的损失函数中,看到继承了nn.Module,并且其中两个参数不像c++那里指定类型,那么他们的类型是哪里来的
这里就是在探索这样一件事
操作逻辑:
N = box1.size(0) M = box2.size(0)
작업 논리:
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))
위 내용은 Python 클래스 매개변수 정의 및 데이터 확장 방법 unsqueeze/expand의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!