Commit 88075146 authored by Sugon_ldc's avatar Sugon_ldc
Browse files

modify readme file data root

parent 5b32a1fe
......@@ -98,7 +98,7 @@ pip3 install -r conda_requirement.txt
pip3 install -e .
```
之后在训练脚本中--cfg-options后面修改data_root值和train,val,test中ann_file的路径
数据集的路径可以通过修改configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py文件中的数据集路径
### 单机多卡
```
......
......@@ -168,21 +168,21 @@ data = dict(
train=dict(
type='BottomUpCocoDataset',
ann_file=f'{data_root}/annotations/person_keypoints_train2017.json',
img_prefix=f'{data_root}/train2017/',
img_prefix=f'{data_root}/images/train2017/',
data_cfg=data_cfg,
pipeline=train_pipeline,
dataset_info={{_base_.dataset_info}}),
val=dict(
type='BottomUpCocoDataset',
ann_file=f'{data_root}/annotations/person_keypoints_val2017.json',
img_prefix=f'{data_root}/val2017/',
img_prefix=f'{data_root}/images/val2017/',
data_cfg=data_cfg,
pipeline=val_pipeline,
dataset_info={{_base_.dataset_info}}),
test=dict(
type='BottomUpCocoDataset',
ann_file=f'{data_root}/annotations/person_keypoints_val2017.json',
img_prefix=f'{data_root}/val2017/',
img_prefix=f'{data_root}/images/val2017/',
data_cfg=data_cfg,
pipeline=test_pipeline,
dataset_info={{_base_.dataset_info}}),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment