update readme

Author: zR

README.md

@@ -12,7 +12,8 @@
 
 **```注意```**
 
-+ 1.如果你需要下载最新版本的代码，请克隆```dev```分支，最新的代码无法保证性能稳定。<br>
++ 如果你需要下载仅有识别的代码，请克隆```only-detect```分支，最新的代码无法保证性能稳定。<br>
++ 如果你正在寻找雷达小目标识别的代码，请前往[这里](https://github.com/zRzRzRzRzRzRzR/Mult-YOLO-alogorithm-of-RoboMaster-Radar-Detection-2023)
 
 ## 环境配置
 
@@ -209,14 +210,12 @@ __数据集开源地址: 链接: https://pan.baidu.com/s/1ayRI1MMw40ae4kuFZCXK_Q
 + 如果推理图像修改，工程中的anchor和图像大小也需要修改，anchor修改请执行```pre-processing_script/change_anchor.py```
 
 ## 总结
-
-+ 该项目针对RMUC2023赛季，如果你有好的建议，欢迎给我留言哦。
-+ 如果你觉得项目对你有帮助，请给个star吧~
-+ 如果你有更好的建议，欢迎提出PR，或者直接联系我哦，大家一起学习！
++ 该项目针对RMUC2023赛季，感谢大家，由于本人已经退役，本项目不在维护。
++ 如果你对该项目有兴趣，可以联系我们团队。
 + __感谢西浦GMaster战队在比赛期间标注数据的嵌入式组和机械组的战友，你们都为视觉组做出了一份重要的贡献！__
 
 ## 联系方式
 
 + 作者微信: zR_ZYX
 + 作者邮箱: [email protected]
-+ 团队邮箱: [email protected]
++ 团队邮箱: [email protected]

pre-processing_script/__init__.py

@@ -1,19 +1,19 @@
-'''
-使用前必看:
-1.本脚本将yolov7训练的聚类环节单独抽出，方便在更改训练和检测数据集大小时修改anchoer的数据，本脚本采用的是yolo自带的k-means聚类方法。
-2.具体参数如下：
-    dataset: 数据的yaml路径
-    n: 类簇的个数
-    img_size: 训练过程中的图片尺寸（32的倍数）
-    thr: anchor的长宽比阈值，将长宽比限制在此阈值之内
-    gen: k - means算法最大迭代次数（不理解的可以去看k - means算法）
-    verbose: 打印参数
-3.输出anchor后，需要将其修改至 data 文件夹下对应的训练cfg文件才能生效。
-'''
-import sys
-sys.path.append('./')
-import utils.autoanchor as autoAC
-
-new_anchors = autoAC.kmean_anchors('../data/winmill/win_kpt.yaml', 9, 416, 5.0, 1000, True)
-print("生成的anchor如下:")
-print(new_anchors)
+'''
+使用前必看:
+1.本脚本将yolov7训练的聚类环节单独抽出，方便在更改训练和检测数据集大小时修改anchoer的数据，本脚本采用的是yolo自带的k-means聚类方法。
+2.具体参数如下：
+    dataset: 数据的yaml路径
+    n: 类簇的个数
+    img_size: 训练过程中的图片尺寸（32的倍数）
+    thr: anchor的长宽比阈值，将长宽比限制在此阈值之内
+    gen: k - means算法最大迭代次数（不理解的可以去看k - means算法）
+    verbose: 打印参数
+3.输出anchor后，需要将其修改至 data 文件夹下对应的训练cfg文件才能生效。
+'''
+import sys
+sys.path.append('./')
+import utils.autoanchor as autoAC
+
+new_anchors = autoAC.kmean_anchors('/root/Yolov7_radar/data/armor-2.yaml', 24, 1280, 5.0, 1000, True)
+print("生成的anchor如下:")
+print(new_anchors)

pre-processing_script/change_anchor.py

@@ -2,55 +2,41 @@
 使用前必看:
 1. 如果你是按照READ.md文档的顺序标注的，那么正常你是用不到这个文件的。如果你是从其他开源站获得的数据，由于标签不对需要更改标签，那么你可以使用这
 个脚本帮助修改。
-2.使用本脚本前请备份你的标注文件。
+2.使用本脚本前请备份你的标注文件。下面的样例是将以YOLO格式导出的LabelStudio数据集转换为西浦GMaster训练的数据集
 '''
 import os
 import argparse
 
 
-def change_label_name(path):
-    total_txt = os.listdir(path)
-    for file in total_txt:
-        file_name = path + '/' + file
-        file = open(file_name, 'r')
-        lines = file.readlines()
-        for index, line in enumerate(lines):
-            t = lines[index]  # 读取当前行的内容
-            num = int(t[0:2])
-
-            # # 在这里修改你想txt操作的内容
-            # if num > 7 and num < 10:
-            #     t = str(num - 1) + t[1:]  # 改成2加字符第二位往后
-            #     lines[index] = t  # 改写lines中的内容
-            if num == 1:
-                t = str(0) + t[1:]  # 改成2加字符第二位往后
-                lines[index] = t  # 改写lines中的内容
-            if num == 8:
-                t = str(1) + t[1:]  # 改成2加字符第二位往后
-                lines[index] = t  # 改写lines中的内容
-            #
-            # if num != 15:  # 切片判断前两个字符
-            #     t = ''
-            # lines[index] = t
-            #
-            # if num > 10:  # 切片判断前两个字符
-            #     t = str(num - 1) + t[2:]
-            #     lines[index] = t
-
-            # if num !=1 and num!=8:
-            #     t = ''
-            #     lines[index] = t
-
-        file.close()
-        t = "".join(lines)
-        file = open(file_name, 'w')
-        file.write(t)
-        file.close()
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('path_dir', type=str, default='', help='Location of path')
+def update_labels(folder_path):
+    # 你的旧的和新的类别顺序
+    old_order = ["B1", "B2", "B3", "B4", "B5", "BB", "BO", "BS", "R1", "R2", "R3", "R4", "R5", "RB", "RO", "RS"]
+    new_order = ["B1", "B2", "B3", "B4", "B5", "BO", "BS", "R1", "R2", "R3", "R4", "R5", "RO", "RS", "BB", "RB"]
+
+    # 创建一个字典来映射旧的id到新的id
+    id_mapping = {old_order.index(name): new_order.index(name) for name in old_order}
+
+    # 遍历标签文件
+    for root, dirs, files in os.walk(folder_path):
+        for file in files:
+            if file.endswith(".txt"):  # 假设标签文件是.txt格式
+                with open(os.path.join(root, file), "r") as f:
+                    lines = f.readlines()
+
+                # 修改每一行的标签id
+                for i in range(len(lines)):
+                    parts = lines[i].split(" ")
+                    parts[0] = str(id_mapping[int(parts[0])])  # 更新id
+                    lines[i] = " ".join(parts)
+
+                # 将修改后的行写回文件
+                with open(os.path.join(root, file), "w") as f:
+                    f.writelines(lines)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Update label files")
+    parser.add_argument("folder_path", type=str, help="Path to the folder containing label files")
     args = parser.parse_args()
-    change_label_name(args.path_dir)
-    print("Finish change_label_id!")
+
+    update_labels(args.folder_path)

pre-processing_script/change_json2yolo.py

@@ -1,33 +1,33 @@
-'''
-使用前必看:
-1. 本代码用来重新批量命名图片和标签。默认状态下是根据命名方式为四位数的文件进行命名，例如将9000.png更改为1.png。
-2. 如果你的原始图片不符合（1）中的命名方式，可以更改 file_list中的key参数进行修改。
-3. 本代码一次只能处理一个文件夹。
-'''
-
-import os
-import re
-import sys
-import argparse
-
-
-@staticmethod
-def rename(path, start_id):
-    file_list = sorted(os.listdir(path), key=lambda x: int(x[:-4]))  # 在这里修改排序规则
-    os.chdir(path)
-    for fileName in file_list:
-        pat = ".+\.(jpg|png|gif|py|txt)"
-        pattern = re.findall(pat, fileName)
-        os.rename(fileName, (str(start_id) + '.' + pattern[0]))
-        start_id = start_id + 1
-    os.chdir(os.getcwd())
-    sys.stdin.flush()
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--path_dir', type=str, default='/media/zr/Data/RoboMaster_data/RMvideos/radar_record_2023/pic', help='Location of path')
-    parser.add_argument('--start_id', type=int, default=6300, help='start sort with the id')
-    args = parser.parse_args()
-    rename(args.path_dir, args.start_id)
-    print("Finish Rename!")
+'''
+使用前必看:
+1. 本代码用来重新批量命名图片和标签。默认状态下是根据命名方式为四位数的文件进行命名，例如将9000.png更改为1.png。
+2. 如果你的原始图片不符合（1）中的命名方式，可以更改 file_list中的key参数进行修改。
+3. 本代码一次只能处理一个文件夹。
+'''
+
+import os
+import re
+import sys
+import argparse
+
+
+@staticmethod
+def rename(path, start_id):
+    file_list = sorted(os.listdir(path), key=lambda x: int(x[:-4]))  # 在这里修改排序规则
+    os.chdir(path)
+    for fileName in file_list:
+        pat = ".+\.(jpg|png|gif|py|txt)"
+        pattern = re.findall(pat, fileName)
+        os.rename(fileName, (str(start_id) + '.' + pattern[0]))
+        start_id = start_id + 1
+    os.chdir(os.getcwd())
+    sys.stdin.flush()
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('path_dir', type=str, default='', help='Location of path')
+    parser.add_argument('--start_id', type=int, default=1, help='start sort with the id')
+    args = parser.parse_args()
+    rename(args.path_dir, args.start_id)
+    print("Finish Rename!")