docs(readme): update <Data>

zjykzj · Jun 29, 2024 · a437553 · a437553
1 parent 8ce77b5
commit a437553
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -72,6 +72,7 @@
 - [Background](#background)
 - [Installation](#installation)
 - [Usage](#usage)
+  - [Data](#data)
   - [Train](#train)
   - [Eval](#eval)
   - [Predict](#predict)
@@ -100,6 +101,15 @@ docker run -it --runtime nvidia --gpus=all --shm-size=16g -v /etc/localtime:/etc
 
 ## Usage
 
+### Data
+
+Download the WIDERFACE dataset from http://shuoyang1213.me/WIDERFACE/, Then convert WIDERFACE dataset format.
+
+```shell
+python3 widerface2yolo.py ../datasets/widerface/WIDER_train/images ../datasets/widerface/wider_face_split/wider_face_train_bbx_gt.txt ../datasets/widerface/
+python3 widerface2yolo.py ../datasets/widerface/WIDER_val/images ../datasets/widerface/wider_face_split/wider_face_val_bbx_gt.txt ../datasets/widerface/
+```
+
 ### Train
 
 ```shell

diff --git a/README.zh-CN.md b/README.zh-CN.md
@@ -72,6 +72,7 @@
 - [背景](#背景)
 - [安装](#安装)
 - [用法](#用法)
+  - [数据](#数据)
   - [训练](#训练)
   - [评估](#评估)
   - [预测](#预测)
@@ -100,6 +101,15 @@ docker run -it --runtime nvidia --gpus=all --shm-size=16g -v /etc/localtime:/etc
 
 ## 用法
 
+### 数据
+
+在[官网](http://shuoyang1213.me/WIDERFACE/)下载WIDERFACE数据集，然后转换数据集格式。
+
+```shell
+python3 widerface2yolo.py ../datasets/widerface/WIDER_train/images ../datasets/widerface/wider_face_split/wider_face_train_bbx_gt.txt ../datasets/widerface/
+python3 widerface2yolo.py ../datasets/widerface/WIDER_val/images ../datasets/widerface/wider_face_split/wider_face_val_bbx_gt.txt ../datasets/widerface/
+```
+
 ### 训练
 
 ```shell