代码仓库为https://github.com/bhsh0112/Script，本仓库以个人应用为导向，总结学习与开发过程中应用到的脚本
1 概述（文件结构）
这里只提供一个功能概述，具体使用方法见下文
| 脚本 | 功能 | 脚本链接 |
|---|---|---|
| split_train_val.py | 划分数据集 | 链接 |
| write_img_path.py | 按要求把文件路径保存到指定文件（yolo用） | 链接 |
| json_to_yolo.py | 对标注文件进行json到yolo的格式转换（yolo用） | 链接 |
| split_files.py | 把一个文件夹内的同类型文件等分到n个子文件夹中（常用于分配任务） | 链接 |
| URL2mp4.py | 下载链接中的视频（当前支持youtube和bilibili） | 链接 |
| Images_download.py | 批量下载链接中的图片 | 链接 |
2 脚本使用说明
2.1 split_train_val.py
作用：划分数据集
代码运行：
1 | python split_train_val.py |
2.2 write_img_path.py
作用：按要求把文件路径保存到指定文件
前期准备：
- 按照格式修改64行的路径
1 | python write_img_path.py |
2.3 json_to_yolo.py
作用：对标注文件进行格式转换
前期准备：
- 按照格式修改55行对检测类别的列举
代码运行：
1 | python json_to_yolo.py |
2.4 split_files.py
作用：把一个文件夹内的同类型文件等分到n个子文件夹中（常用于分配任务）
前期准备：需要修改代码参数
- 参数说明
- source_dir：要分配的文件夹
- file_extension：要分配文件的扩展名，例：.txt
- num_folders：要等分成几份
代码运行：
1 | python split_files.py --source_dir /path/to/folder --file_extension .[extension] --num_folders n |
例：
1 | python split_files.py --source_dir images file_extension .jpg --num_folders 5 |
2.5 URL2mp4.py
作用：输入视频网站链接（当前支持bilibili和youtube），即可在当前目录下新建一个Downloads文件夹，保存链接中的视频
前期准备：根据以下说明配置环境
python -m pip install yt-dlp- 安装ffmpeg（不同系统或不同，可自查教程，通常单一指令即可）
代码运行：
1 | python URL2mp4.mp4 |
运行后在终端输入URL即可
2.6 images_download.py
作用：输入网站链接及保存路径，即可将网站链接中的所有图片保存到输入的保存路径下
前期准备：根据以下说明配置环境
1 | pip install urllib3 |
代码运行：
1 | python images_download.py |
运行后输入URL及保存地址