Subject | Hash | Author | Date (UTC) |
---|---|---|---|
wip: pacnn | 2192d7c7b449fecf3868877d9cfbc09bb6f7ae98 | Thai Thien | 2019-09-22 03:44:56 |
wip: pacnn | 37620e5a9bc0f9516ea964ec58d9bdaa1c40ff36 | Thai Thien | 2019-09-22 03:14:42 |
fix training flow | 2b87b1b26c7296b64493fdc49fedb421b249dfa3 | Thai Thien | 2019-09-17 18:00:35 |
dataset script | bc5c052f5f956510ab95ef9a45434fd486c57fae | Thai Thien | 2019-09-16 17:21:13 |
evaluator | ffc5bf8290ae0c469a9a18a2d061cfd1bfeee822 | Thai Thien | 2019-09-14 04:56:35 |
some more test for data loader | 25173578cde7d4e9fe6c6140d1ee01caa4fcfc32 | Thai Thien | 2019-09-14 02:51:58 |
some visualize to debug data loader | e4f52007616acf307bddbde79c0fb4f8c649c785 | Thai Thien | 2019-09-13 17:35:45 |
wip | d7d44cad6774355bdfa45414258763f6c6a0c299 | Thai Thien | 2019-08-31 16:58:16 |
commit all | 6dad7a58f7dbf9fc288ce9dd3e92be538851c2a7 | Thai Thien | 2019-08-29 19:10:44 |
input d1,d2,d3 match | fc2a809241f8b6356d964c63d40cbebd55ca5f6c | Thai Thien | 2019-08-28 17:57:05 |
WIP | 39eab26d061e61dfffbf164dbd5fd878299b7250 | thient | 2019-08-28 11:09:12 |
output of de is ok | dd770386674df3e0fbebafdfc48a9352bc28967d | thient | 2019-08-28 10:54:09 |
code pacnn | c49537b5cc91e96e4e35c9338d2c95b9bb41c672 | Thai Thien | 2019-08-27 16:35:27 |
crowd counting stuff | da9f27a39cba9bdd021b6b5c562f5f7c2be50190 | Thai Thien | 2019-08-24 18:27:44 |
seem ok | 53fa176c31669a0e89b04adf290cb398f0316c45 | Thai Thien | 2019-08-24 18:26:31 |
flow ok | ad849681000818dfbcd0c1715c2858aed7236041 | Thai Thien | 2019-08-24 17:00:02 |
wip | 23c3ec48497782bbc91d829e1c8a682502360ab9 | Thai Thien | 2019-08-24 14:19:22 |
work in progress, try to use https://pytorch.org/ignite/quickstart.html | 39c824fe8fc2501628ee42c236a844df45521007 | Thai Thien | 2019-08-24 07:41:46 |
Work in progress | 984be31d85e5cbdb2af296ccdb128381fe9bf09e | Thai Thien | 2019-08-24 05:30:51 |
README | 045706df1fa3452d150a190675c60e80ebd18e08 | Thai Thien | 2019-08-24 04:11:49 |
File | Lines added | Lines deleted |
---|---|---|
models/pacnn.py | 5 | 3 |
File models/pacnn.py changed (mode: 100644) (index a16141c..6ccb98e) | |||
... | ... | class PACNNWithPerspectiveMap(nn.Module): | |
58 | 58 | # deconvolution upsampling | # deconvolution upsampling |
59 | 59 | self.up12 = nn.ConvTranspose2d(512, 1, 2, 2) | self.up12 = nn.ConvTranspose2d(512, 1, 2, 2) |
60 | 60 | self.up23 = nn.ConvTranspose2d(512, 1, 2, 2) | self.up23 = nn.ConvTranspose2d(512, 1, 2, 2) |
61 | self.up_perspective = nn.ConvTranspose2d(512, 1, 2, 2) | ||
61 | 62 | ||
62 | 63 | # if true, use perspective aware | # if true, use perspective aware |
63 | 64 | # if false, use average | # if false, use average |
... | ... | class PACNNWithPerspectiveMap(nn.Module): | |
68 | 69 | de2 = self.de2_11((self.de2net(x))) | de2 = self.de2_11((self.de2net(x))) |
69 | 70 | de3 = self.de3_11((self.de3net(x))) | de3 = self.de3_11((self.de3net(x))) |
70 | 71 | if self.perspective_aware_mode: | if self.perspective_aware_mode: |
71 | respective = self.perspective_11(self.perspective_net) | ||
72 | pespective_w_s = self.perspective_11(self.perspective_net(x)) | ||
73 | pespective_w = self.up_perspective(pespective_w_s) | ||
72 | 74 | # TODO: code more here | # TODO: code more here |
75 | de23 = pespective_w_s * de2 + (1 - pespective_w_s)*(de2 + self.up23(de3)) | ||
76 | de = pespective_w * de1 + (1 - pespective_w)*(de2 + self.up12(de23)) | ||
73 | 77 | else: | else: |
74 | 78 | de23 = (de2 + self.up23(de3))/2 | de23 = (de2 + self.up23(de3))/2 |
75 | 79 | de = (de1 + self.up12(de23))/2 | de = (de1 + self.up12(de23))/2 |
... | ... | def parameter_count_test(): | |
91 | 95 | sum_of_part = backbone + de1_11 + de2_11 + de3_11 + conv611 | sum_of_part = backbone + de1_11 + de2_11 + de3_11 + conv611 |
92 | 96 | print(sum_of_part) | print(sum_of_part) |
93 | 97 | ||
94 | |||
95 | |||
96 | 98 | if __name__ == "__main__": | if __name__ == "__main__": |
97 | 99 | parameter_count_test() | parameter_count_test() |
98 | 100 | # net = PACNN() | # net = PACNN() |