From: @wukesong Reviewed-by: @wuxuejian,@oacjiewen Signed-off-by: @wuxuejiantags/v1.2.0-rc1
| @@ -139,7 +139,7 @@ The structure of the files in this repository is shown below. | |||||
| │ ├─ network_define.py // define loss | │ ├─ network_define.py // define loss | ||||
| │ └─ predict.py // predict keypoints from heatmaps | │ └─ predict.py // predict keypoints from heatmaps | ||||
| ├─ eval.py // evaluation script | ├─ eval.py // evaluation script | ||||
| ├─ param_convert.py // model parameters convertion script | |||||
| ├─ param_convert.py // model parameters conversion script | |||||
| ├─ train.py // training script | ├─ train.py // training script | ||||
| └─ README.md // descriptions about this repository | └─ README.md // descriptions about this repository | ||||
| ``` | ``` | ||||
| @@ -47,7 +47,7 @@ class KeypointDatasetGenerator: | |||||
| self.rotation_factor = cfg.DATASET.ROT_FACTOR | self.rotation_factor = cfg.DATASET.ROT_FACTOR | ||||
| self.flip = cfg.DATASET.FLIP | self.flip = cfg.DATASET.FLIP | ||||
| # dataset informations | |||||
| # dataset information | |||||
| self.db = [] | self.db = [] | ||||
| self.is_train = is_train | self.is_train = is_train | ||||
| @@ -55,7 +55,7 @@ Tiny-DarkNet是Joseph Chet Redmon等人提出的一个16层的针对于经典的 | |||||
| <!-- # [Features](#contents) | <!-- # [Features](#contents) | ||||
| ## [Distrubuted](#contents) | |||||
| ## [Distributed](#contents) | |||||
| <!-- 不同的机器有同一个模型的多个副本,每个机器分配到不同的数据,然后将所有机器的计算结果按照某种方式合并 --> | <!-- 不同的机器有同一个模型的多个副本,每个机器分配到不同的数据,然后将所有机器的计算结果按照某种方式合并 --> | ||||
| @@ -73,7 +73,7 @@ if __name__ == '__main__': | |||||
| if args_opt.dataset_name == "imagenet": | if args_opt.dataset_name == "imagenet": | ||||
| cfg = imagenet_cfg | cfg = imagenet_cfg | ||||
| else: | else: | ||||
| raise ValueError("Unsupport dataset.") | |||||
| raise ValueError("Unsupported dataset.") | |||||
| # set context | # set context | ||||
| device_target = cfg.device_target | device_target = cfg.device_target | ||||
| @@ -97,7 +97,7 @@ if __name__ == '__main__': | |||||
| if args_opt.dataset_name == "imagenet": | if args_opt.dataset_name == "imagenet": | ||||
| dataset = create_dataset_imagenet(cfg.data_path, 1) | dataset = create_dataset_imagenet(cfg.data_path, 1) | ||||
| else: | else: | ||||
| raise ValueError("Unsupport dataset.") | |||||
| raise ValueError("Unsupported dataset.") | |||||
| batch_num = dataset.get_dataset_size() | batch_num = dataset.get_dataset_size() | ||||