Skip to content

Commit 37f03a7

Browse files
authored
fixed some issues (wang-xinyu#400)
* 1 动态输入 2 图像预处理和后处理和pytorch同步 3 代码格式统一 * 重新梳理代码 * tab 替换为4 space * HRNet-Semantic-Segmentation * update result img * update readme * Synchronize * Modify some bugs
1 parent b15a45d commit 37f03a7

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

dbnet/common.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,8 @@ ILayer* convBnLeaky(INetworkDefinition *network, std::map<std::string, Weights>&
110110
conv1->setPaddingNd(DimsHW{ p, p });
111111
conv1->setNbGroups(g);
112112
//IScaleLayer* bn1 = addBatchNorm2d(network, weightMap, *conv1->getOutput(0), lname + ".bn", 1e-4);
113-
IScaleLayer* bn1 = addBatchNorm2d(network, weightMap, *conv1->getOutput(0), lname.substr(0, lname.find_last_of(".")) + bnname, 1e-3);
114-
auto lr = network->addActivation(*bn1->getOutput(0), ActivationType::kLEAKY_RELU);
113+
IScaleLayer* bn1 = addBatchNorm2d(network, weightMap, *conv1->getOutput(0), lname.substr(0, lname.find_last_of(".")) + bnname, 1e-5);
114+
auto lr = network->addActivation(*bn1->getOutput(0), ActivationType::kRELU);
115115
lr->setAlpha(0.1);
116116
return lr;
117117
}

hrnet/hrnet-semantic-segmentation/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
# HRNet-Semantic-Segmentation
22

3-
The Pytorch implementation is [HRNet-Semantic-Segmentation](https://github.com/HRNet/HRNet-Semantic-Segmentation). The implemented model is **HRNetV2-W18-Small-v2**
3+
The Pytorch implementation is [HRNet-Semantic-Segmentation-v1.1](https://github.com/HRNet/HRNet-Semantic-Segmentation/tree/pytorch-v1.1). The implemented model is **HRNetV2-W18-Small-v2**
44

55

66
## How to Run
77

88
* 1. generate .wts
99

10-
Download code and model from [HRNet-Semantic-Segmentation](https://github.com/HRNet/HRNet-Semantic-Segmentation) and config your environments.
10+
Download code and model from [HRNet-Semantic-Segmentation-v1.1](https://github.com/HRNet/HRNet-Semantic-Segmentation/tree/pytorch-v1.1) and config your environments.
1111

1212
Put `demo.py` in the `YOUR_ROOT_DIR\HRNet-Semantic-Segmentation\tools ` folder, set `savewts in main()` as `True`, and run, the .wts will be generated.
1313

0 commit comments

Comments
 (0)