添加状态分类和液面分割
This commit is contained in:
@ -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" # 保存最终数据集的根目录
|
||||
|
||||
Reference in New Issue
Block a user