save_dir = increment_path(Path(project) / name, exist_ok=exist_ok) # increment run (save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make dir实现根据读取的文件夹在lables下创建相同的文件夹名称
时间: 2024-04-12 15:28:04 浏览: 170
要实现根据读取的文件夹在 `labels` 下创建相同的文件夹名称,您可以进行以下修改:
```python
save_dir = increment_path(Path(project) / name, exist_ok=exist_ok) # increment run
if save_txt:
(save_dir / 'labels').mkdir(parents=True, exist_ok=True) # create labels folder
# Read the folder name from the input folder path
folder_name = os.path.basename(source_folder_name)
(save_dir / 'labels' / folder_name).mkdir(parents=True, exist_ok=True) # create folder with the same name under labels
```
在这个修改后的代码中,我们首先判断 `save_txt` 变量是否为 `True`。如果是,我们在 `save_dir` 下创建一个名为 `'labels'` 的文件夹。
然后,使用 `os.path.basename(source_folder_name)` 提取输入文件夹路径的文件夹名称,并将其保存到 `folder_name` 变量中。
最后,我们在 `save_dir / 'labels'` 下创建一个与读取的文件夹名称相同的文件夹。
这样修改后,您将能够根据读取的文件夹在 `labels` 下创建相同的文件夹名称。
阅读全文