把mnist文件转成yolov5可以训练的文件

将MNIST数据集转换为Yolo格式需要进行以下步骤：

1. 将MNIST数据集的图像和标签转换为Yolo格式的标注文件。
2. 将Yolo格式的标注文件转换为Yolo需要的训练文件。

下面是一个简单的示例代码：

import os
import numpy as np
from PIL import Image

# 标签列表
labels = ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"]

# MNIST数据集路径
mnist_path = "path/to/mnist/"

# Yolo格式的标注文件路径
labels_path = "path/to/labels.txt"

# Yolo格式的训练文件路径
train_path = "path/to/train.txt"

# 打开标注文件
with open(labels_path, "w") as labels_file:
    # 遍历MNIST数据集
    for i in range(60000):
        # 加载图像和标签
        img_path = os.path.join(mnist_path, "train", str(i) + ".png")
        label_path = os.path.join(mnist_path, "train-labels", str(i) + ".txt")
        img = Image.open(img_path)
        label = np.loadtxt(label_path)

        # 转换为Yolo格式的标注
        label_yolo = []
        for j in range(label.shape[0]):
            x_center = label[j, 0] + label[j, 2] / 2
            y_center = label[j, 1] + label[j, 3] / 2
            w = label[j, 2]
            h = label[j, 3]
            label_yolo.append(labels.index(str(int(label[j, 4]))) + 1)
            label_yolo.append(x_center / img.width)
            label_yolo.append(y_center / img.height)
            label_yolo.append(w / img.width)
            label_yolo.append(h / img.height)
        label_yolo = [str(l) for l in label_yolo]

        # 写入标注文件
        labels_file.write(" ".join(label_yolo) + "\n")

        # 写入训练文件
        train_file = os.path.join(mnist_path, "train.txt")
        with open(train_file, "a") as f:
            f.write(os.path.abspath(img_path) + "\n")

这将生成一个名为“train.txt”的文件，其中包含MNIST数据集图像的绝对路径，以及一个名为“labels.txt”的文件，其中包含MNIST数据集图像的Yolo格式标注。注意，这只是一个简单的示例代码，你可能需要根据自己的需求进行修改。