hackiey / keypoints

A pytorch implementation of Google's paper <Towards Accurate Multi-person Pose Estimation in the Wild>
MIT License
102 stars 25 forks source link

question about the model.py #3

Open Liz66666 opened 5 years ago

Liz66666 commented 5 years ago

Hi @hackiey , Thanks for your code. But I still have a question about the network in your implement. Paper said that "we employ atrous convolution to generate the 3 · K predictions", but in your model.py, you used nn.ConvTranspose2d. Is these two methods same? https://github.com/hackiey/keypoints/blob/a676ef08ee8f32d68d2fa1273955cf3ff7b5f9f4/src/model.py#L24

HowieMa commented 4 years ago

In line 10: self.num_outputs = num_classes * 3