用pytorch写nn.Linear输出五个坐标值和边框大小并训练的代码

以下是一个简单的 PyTorch 代码示例，展示如何使用 nn.Linear 输出五个坐标值和边框大小，并进行训练：

import torch
import torch.nn as nn
import torch.optim as optim

# 定义模型
class MyModel(nn.Module):
    def __init__(self, input_size):
        super(MyModel, self).__init__()
        self.linear = nn.Linear(input_size, 5)

    def forward(self, x):
        return self.linear(x)

# 创建模型
input_size = 64
model = MyModel(input_size)

# 定义损失函数和优化器
criterion = nn.MSELoss()
optimizer = optim.SGD(model.parameters(), lr=0.01)

# 创建输入和目标张量
batch_size = 32
input_tensor = torch.randn(batch_size, input_size)
target_tensor = torch.randn(batch_size, 5)

# 训练模型
for epoch in range(100):
    # 前向传播
    output_tensor = model(input_tensor)

    # 计算损失
    loss = criterion(output_tensor, target_tensor)

    # 反向传播
    optimizer.zero_grad()
    loss.backward()
    optimizer.step()

    # 打印损失
    print('Epoch %d, Loss: %.4f' % (epoch, loss.item()))

这个代码示例中，我们首先定义了一个 MyModel 类，它包含一个 nn.Linear 层，用于将输入张量映射到输出张量。然后我们创建了模型对象 model，定义了损失函数 criterion 和优化器 optimizer。接着我们创建了输入张量 input_tensor 和目标张量 target_tensor。在训练过程中，我们将输入张量输入到模型中，得到输出张量 output_tensor，然后计算损失，进行反向传播和优化。最后我们打印了每个 epoch 的损失。

需要注意的是，这个示例中的目标张量 target_tensor 是随机生成的，实际应用中需要根据具体任务设置目标张量。另外，这里使用了均方误差损失函数，适用于回归任务，如果是分类任务需要使用交叉熵损失函数等适合的损失函数。