labeling下批量修改标签

最新推荐文章于 2024-04-09 23:16:58 发布

weixin_48239054

最新推荐文章于 2024-04-09 23:16:58 发布

阅读量2k

点赞数

文章标签： python c#

本文链接：https://blog.csdn.net/weixin_48239054/article/details/123753127

版权

用labeling标记图片，是一件工作量非常大的事情，需要人一张张图把需要识别的物品标识出来。所以一般都会让几个人一起配合标记。

但标记过程中，由于沟通问题，导致标记错误，这时候就需要修改标签了。

首先，我们得知道labeling保存文件格式是怎么样的：

可以看到0是标签，而后面四位是两个和坐标相关的值

我们要做的就是统一修标签数字

这里上一份用C#写的小程序

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace WindowsFormsApplication1
{
    public partial class Form1 : Form
    {
        Dictionary<string, string> dictionary = new Dictionary<string, string>();
        void newDictionary()
        {
            string[] allNum = textBox1.Text.Split(new string[] { " " }, StringSplitOptions.RemoveEmptyEntries);
            dictionary.Clear();
            foreach(string str in allNum)
            {
                if (!string.IsNullOrEmpty(str) && str.Contains("-"))
                {
                    string[] index = str.Split(new string[] { "-" }, StringSplitOptions.RemoveEmptyEntries);
                    if (index.Length == 2 && !string.IsNullOrEmpty(index[0]) && !string.IsNullOrEmpty(index[1]))
                    {
                        dictionary.Add(index[0], index[1]);
                    }
                }
            }
        }
        public Form1()
        {
            InitializeComponent();
            textBox2.Text = AppDomain.CurrentDomain.BaseDirectory + "Annotations";
        }

        private void button1_Click(object sender, EventArgs e)
        {
            newDictionary();
            DirectoryInfo dir = new DirectoryInfo(textBox2.Text);
            FileInfo[] files = dir.GetFiles("*.txt");
            foreach (FileInfo file in files)
            {
                if (file.Name != "classes.txt")
                    ReadTxtContent(file.FullName.ToString(), file.Name);
            }
        }

        /// <summary>
        /// 读取txt文件内容
        /// </summary>
        /// <param name="Path">文件地址</param>
        /// <param name="name">文件名</param>
        public void ReadTxtContent(string Path,string name)
        {
            StreamReader sr = new StreamReader(Path, Encoding.Default);
            string content;
            List<string> end=new List<string>();
            string end2 = "";
            while ((content = sr.ReadLine()) != null)
            {
                end.Add(content);
            }
            foreach(string end_ in end)
            {
                string[] data = end_.Split(new string[] { " " }, StringSplitOptions.RemoveEmptyEntries);
                if (data.Length == 5 && !string.IsNullOrEmpty(data[0]) && dictionary.ContainsKey(data[0]))
                {
                    data[0] = dictionary[data[0]];//修改开头数字
                    foreach (string str in data)
                    {
                        end2 += str + " ";
                    }
                    end2 = end2.Remove(end2.Length - 1, 1);//删除最后的空格
                    WriteTxt(end2, name);
                    end2 = "";
                }
            }
        }

        /// <summary>
        /// 生成修改后文件
        /// </summary>
        /// <param name="content">生成文件的内容</param>
        /// <param name="name">生成文件的文件名</param>
        public static void WriteTxt(string content,string name)
        {
            string path = AppDomain.CurrentDomain.BaseDirectory;
            if (!string.IsNullOrEmpty(path))
            {
                path = AppDomain.CurrentDomain.BaseDirectory + "out";
                if (!Directory.Exists(path))
                {
                    Directory.CreateDirectory(path);
                }
                path = path + "\\" + name;
                if (!File.Exists(path))
                {
                    FileStream fs = File.Create(path);
                    fs.Close();
                }
                if (File.Exists(path))
                {
                    StreamWriter sw = new StreamWriter(path, true, System.Text.Encoding.Default);
                    sw.WriteLine(content);
                    sw.Close();
                }
            }
        }
    }
}