添加状态分类和液面分割

This commit is contained in:
琉璃月光
2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions

View File

@ -77,7 +77,7 @@ def split_dataset(
if __name__ == "__main__":
# 数据集路径
images_dir = r"/home/hx/桌面/image/image" # 需要读取的所有图像文件夹路径
labels_dir = r"/home/hx/桌面/image/2" # 需要读取的所有图像与之对应的txt标签文件夹路径
labels_dir = r"/home/hx/桌面/image/image/2" # 需要读取的所有图像与之对应的txt标签文件夹路径
# 输出路径
output_root = r"/home/hx/桌面/image" # 保存最终数据集的根目录