C#本地将labelme数据集转换为机器视觉yolo数据集格式

C#本地，将labelme数据集转换为机器视觉yolo数据集格式

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.Encodings.Web;
using System.Text.RegularExpressions;
using System.Text.Unicode;
using System.Threading.Tasks;
using System.Windows.Ink;
using System.Windows.Media.Media3D;namespace WpfAppYolo11_test1
{/// <summary>/// 将labelme数据集转换为yolo数据集格式/// </summary>public class LabelmeToYoloDataSetUtils{Dictionary<string, short> labels = new Dictionary<string, short>();/// <summary>/// 转换后保存目录/// </summary>const string outDir = "D:\\yoloTrainDataSet";const int saveLength = 10;/// <summary>/// labelme转yolo格式数据，要转换的目录/// </summary>/// <param name="inputDir"></param>/// 2025-3-4 00:45:38，public void FileConvert(string inputDir){//labels.Clear();var files = System.IO.Directory.GetFiles(inputDir, "*.json");if (files == null || files.Length == 0){return;}//获取目录中文件夹，以数字升序命名最新的训练文件夹var folders = Directory.GetDirectories(outDir);int lastNum = 1;if (folders != null && folders.Length > 0){List<DirectoryInfo> dirList = new List<DirectoryInfo>();foreach (var item in folders){DirectoryInfo directoryInfo = new DirectoryInfo(item);dirList.Add(directoryInfo);}var lastDir = dirList.OrderByDescending(g => g.CreationTime).FirstOrDefault();//获取文件夹后缀var lastNumStr = Regex.Match(lastDir.Name, "\\d*$").Value;lastNum = Convert.ToInt32(lastNumStr) + 1;}//此次训练保存的文件夹string newTrainDir = "train" + lastNum;//string dirNew = "train" + Guid.NewGuid().ToString("n");string dirNew = outDir + "\\" + newTrainDir + "\\train";//处理json文件foreach (var item in files){//处理单个json文件SingleFileConvert(item, dirNew);}//挑选训练图片前两个图片作为验证集//创建val文件夹string val = Path.Combine(outDir, newTrainDir, "val");                      var files21 = System.IO.Directory.GetFiles(inputDir, "*.jpg");var files22 = System.IO.Directory.GetFiles(inputDir, "*.png");var fileContainer = files21.Union(files22).ToList();var first1 = fileContainer[0];var first2 = fileContainer[1];//将训练图片全部复制到train中var trainImgNew = Path.Combine(dirNew, "images");if (!Directory.Exists(trainImgNew)){Directory.CreateDirectory(trainImgNew);}foreach (var itemImg in fileContainer){string fileNew2  = Path.Combine(trainImgNew, Path.GetFileName(itemImg));File.Copy(itemImg, fileNew2);}//只有2个样本时，将训练的文件夹作为验证if (files.Length <= 2){val = dirNew;goto Val_done;}if (!Directory.Exists(val)){Directory.CreateDirectory(val);}//图片文件夹string valImage = Path.Combine(val, "images");if (!Directory.Exists(valImage)){Directory.CreateDirectory(valImage);}string file1Name = Path.GetFileName(first1);string file2Name = Path.GetFileName(first2);string img1 = Path.Combine(valImage, file1Name);File.Copy(first1, img1);string img2 = Path.Combine(valImage, file2Name);File.Copy(first2, img2);//坐标数据string vallabels = Path.Combine(val, "labels");if (!Directory.Exists(vallabels)){Directory.CreateDirectory(vallabels);}//找到train目录中的坐标数据string vallabels2 = Path.Combine(dirNew, "labels");string txt1 = Path.GetFileNameWithoutExtension(file1Name) + ".txt";string txt2 = Path.GetFileNameWithoutExtension(file2Name) + ".txt";string fileLabel1 = Path.Combine(vallabels2, txt1);string fileLabel2 = Path.Combine(vallabels2, txt2);if (File.Exists(fileLabel1)){string fileLabelTxt1 = Path.Combine(vallabels, txt1);File.Copy(fileLabel1, fileLabelTxt1);}if (File.Exists(fileLabel2)){string fileLabelTxt2 = Path.Combine(vallabels, txt2);File.Copy(fileLabel2, fileLabelTxt2);}Val_done://生成data.yamlstring dir = dirNew;//Path.Combine(outDir, newTrainDir);dir = dir.Replace("\\", "/");val = val.Replace("\\", "/");StringBuilder sb = new StringBuilder();sb.AppendLine("train: " + dir);sb.AppendLine("val: " + val);sb.AppendLine("test: " + val);sb.AppendLine("");sb.AppendLine("nc: " + labels.Count);//设置编码支持中文string labelName = System.Text.Json.JsonSerializer.Serialize(labels.Keys,new System.Text.Json.JsonSerializerOptions(){Encoder = JavaScriptEncoder.Create(UnicodeRanges.All)});labelName = labelName.Replace("\"", "'");//替换双引号为单引号sb.AppendLine("names: " + labelName);//保存//var arr = inputDir.Split("\\");//var lastDir = arr.Last();//string newDir = inputDir.TrimEnd(lastDir.ToArray());//string filePath = Path.Combine(newDir, "yoloTrainDataSet", "data.yaml");string yamlDir = outDir + "\\" + newTrainDir;string filePath = Path.Combine(yamlDir, "data.yaml");System.IO.File.WriteAllText(filePath, sb.ToString());}/// <summary>/// 单个json文件转换/// </summary>public void SingleFileConvert(string file, string inputDir){string json = System.IO.File.ReadAllText(file);var data2 = System.Text.Json.JsonSerializer.Deserialize<LabelMeView>(json);if (data2 == null || data2.shapes == null || data2.shapes.Count == 0){return;}StringBuilder stringBuilder = new StringBuilder();//读取每个形状foreach (var item in data2.shapes){//忽略没名称的形状if (string.IsNullOrEmpty(item.label)){continue;}labels.TryAdd(item.label, 1);double x1 = 0, x2 = 0, y1 = 0, y2 = 0;if (item.shape_type == "rectangle"){x1 = item.points[0][0];y1 = item.points[0][1];x2 = item.points[1][0];y2 = item.points[1][1];}else if (item.shape_type == "polygon"){x1 = item.points.Select(g => g[0]).Min();y1 = item.points.Select(g => g[1]).Min();x2 = item.points.Select(g => g[0]).Max();y2 = item.points.Select(g => g[1]).Max();}//中心点double x_center = (x1 + x2) / (double)2 / data2.imageWidth;double y_center = (y1 + y2) / (double)2 / data2.imageHeight;double width = Math.Abs((x2 - x1) / data2.imageWidth);double height = Math.Abs((y2 - y1) / data2.imageHeight);//获取对象idint classId = labels.Keys.ToList().IndexOf(item.label);stringBuilder.AppendLine($"{classId} {x_center} {y_center} {width} {height}");}//保存txt文件//var arr = inputDir.Split("\\");//var lastDir = arr.Last();//string newDir = inputDir.TrimEnd(lastDir.ToArray());// string dir = System.IO.Path.Combine( outDir, "labels");string dir = System.IO.Path.Combine(inputDir, "labels");if (!System.IO.Directory.Exists(dir)){System.IO.Directory.CreateDirectory(dir);}//string fileName = Path.GetFileName(file) + "_" + Guid.NewGuid().ToString("n") + ".txt";string fileName = Path.GetFileNameWithoutExtension(file) + ".txt";string filePath = System.IO.Path.Combine(dir, fileName);File.WriteAllText(filePath, stringBuilder.ToString());}}public class LabelMeView{public string version { get; set; }public List<ShapeView> shapes { get; set; }public string imagePath { get; set; }/// <summary>/// 图片高度,px/// </summary>public int imageHeight { get; set; }/// <summary>/// 图片宽度,px/// </summary>public int imageWidth { get; set; }}public class ShapeView{/// <summary>/// 对象说明，标签/// </summary>public string label { get; set; }/// <summary>/// 形状描述；rectangle；polygon/// </summary>public string shape_type { get; set; }public object flags { get; set; }/// <summary>/// 坐标点/// </summary>public List<double[]> points { get; set; }public string version { get; set; }}}