Skip to content

Commit 5e5f3c3

Browse files
authored
Update README.md
1 parent 530fa0e commit 5e5f3c3

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
11
# High-resolution networks (HRNets) for Image classification
22

3+
## News
4+
- TensoFlow implemenation https://github.com/yuanyuanli85/tf-hrnet. Thanks VictorLi (https://github.com/yuanyuanli85)!
5+
6+
- Enable ONNX export after fixing issue. Thanks Baowen Bao (https://github.com/BowenBao)!
7+
38
## Introduction
49
This is the official code of [high-resolution representations for ImageNet classification](https://arxiv.org/abs/1904.04514).
510
We augment the HRNet with a classification head shown in the figure below. First, the four-resolution feature maps are fed into a bottleneck and the number of output channels are increased to 128, 256, 512, and 1024, respectively. Then, we downsample the high-resolution representations by a 2-strided 3x3 convolution outputting 256 channels and add them to the representations of the second-high-resolution representations. This process is repeated two times to get 1024 channels over the small resolution. Last, we transform 1024 channels to 2048 channels through a 1x1 convolution, followed by a global average pooling operation. The output 2048-dimensional representation is fed into the classifier.

0 commit comments

Comments
 (0)