Skip to content

Commit bb67205

Browse files
readme typo fixed (#343)
Co-authored-by: Ken Park <spark@knowru.com>
1 parent 1c16986 commit bb67205

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docs/markdown/quick_start/training.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ Config.set_model_name(args.model_name)
1818
Config.set_model_type(Config.MODEL.LightweightOpenpose)
1919
Config.set_model_backbone(Config.BACKBONE.Vggtiny)
2020
# >>> set dataset to use
21-
Config.set_dataset_type(Config.DATA.COCO)
21+
Config.set_dataset_type(Config.DATA.MSCOCO)
2222
# >>> set training type
2323
Config.set_train_type(Config.TRAIN.Single_train)
2424
# >>> configuration is done, get config object to assemble the system
@@ -54,7 +54,7 @@ Config.set_model_name(args.model_name)
5454
Config.set_model_type(Config.MODEL.LightweightOpenpose)
5555
Config.set_model_backbone(Config.BACKBONE.Vggtiny)
5656
# >>> set dataset to use
57-
Config.set_dataset_type(Config.DATA.COCO)
57+
Config.set_dataset_type(Config.DATA.MSCOCO)
5858
# >>> configuration is done, get config object to assemble the system
5959
config=Config.get_config()
6060
model=Model.get_model(config)

docs/markdown/tutorial/training.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ Config.set_model_name(args.model_name)
9595
Config.set_model_type(Config.MODEL.LightweightOpenpose)
9696
Config.set_model_backbone(Config.BACKBONE.Vggtiny)
9797
# >>> set dataset to use
98-
Config.set_dataset_type(Config.DATA.COCO)
98+
Config.set_dataset_type(Config.DATA.MSCOCO)
9999
# >>> configuration is done, get config object to assemble the system
100100
config=Config.get_config()
101101
model=Model.get_model(config)
@@ -164,7 +164,7 @@ model, functions of preprocess, postprocess and visualize are provided.
164164
The preprocess function is able to convert the annotaion into targets used for training the model.
165165
```bash
166166
preprocess=Model.get_preprocess(Config.MODEL.LightweightOPenpose)
167-
conf_map,paf_map=preprocess(annos,img_height,img_width,model_hout,model_wout,Config.DATA.COCO,data_format="channels_first")
167+
conf_map,paf_map=preprocess(annos,img_height,img_width,model_hout,model_wout,Config.DATA.MSCOCO,data_format="channels_first")
168168
pd_conf_map,pd_paf_map=my_model.forward(input_image[np.newaxis,...])
169169
my_model.cal_loss(conf_map,pd_conf_map,paf_map,pd_paf_map)
170170
```

0 commit comments

Comments
 (0)