Subject | Hash | Author | Date (UTC) |
---|---|---|---|
fix pacnn avg schema | c2140a96886195782e5689c24aeeb4fe7a2db7ad | Thai Thien | 2019-09-22 17:35:01 |
debug number not divisible by 8 | a568fd7f294a8bd31b3db78437b4b6b51b5b41b9 | Thai Thien | 2019-09-22 04:36:06 |
pacnn | 967074890d14ab0eefc277801860270a468e8f9f | Thai Thien | 2019-09-22 03:54:48 |
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 |
File | Lines added | Lines deleted |
---|---|---|
main_pacnn.py | 5 | 3 |
models/pacnn.py | 13 | 2 |
models/playground.py | 2 | 2 |
File main_pacnn.py changed (mode: 100644) (index 449320c..9024f16) | |||
... | ... | if __name__ == "__main__": | |
126 | 126 | ||
127 | 127 | # evaluate | # evaluate |
128 | 128 | ||
129 | best_checkpoint = torch.load("test2checkpoint.pth.tar") | ||
130 | net = PACNN().to(device) | ||
129 | |||
130 | net = PACNNWithPerspectiveMap(PACNN_PERSPECTIVE_AWARE_MODEL).to(device) | ||
131 | 131 | print(net) | print(net) |
132 | net.load_state_dict(best_checkpoint['state_dict']) | ||
132 | |||
133 | # best_checkpoint = torch.load("test2checkpoint.pth.tar") | ||
134 | # net.load_state_dict(best_checkpoint['state_dict']) | ||
133 | 135 | ||
134 | 136 | # device = "cpu" | # device = "cpu" |
135 | 137 | mae_calculator_d1 = MAECalculator() | mae_calculator_d1 = MAECalculator() |
File models/pacnn.py changed (mode: 100644) (index 9e7203e..46f1e5d) | |||
1 | 1 | import torch.nn as nn | import torch.nn as nn |
2 | 2 | import torch | import torch |
3 | import torch.nn.functional as F | ||
3 | 4 | ||
4 | 5 | from torchvision import models | from torchvision import models |
5 | 6 | import numpy as np | import numpy as np |
... | ... | class PACNNWithPerspectiveMap(nn.Module): | |
78 | 79 | #try: | #try: |
79 | 80 | pespective_w_s = None | pespective_w_s = None |
80 | 81 | pespective_w = None | pespective_w = None |
81 | de23 = (de2 + self.up23(de3))/2 | ||
82 | de = (de1 + self.up12(de23))/2 | ||
82 | upde3 = self.up23(de3) | ||
83 | pad_3_0 = de2.size()[2] - upde3.size()[2] | ||
84 | pad_3_1 = de2.size()[3] - upde3.size()[3] | ||
85 | upde3pad = F.pad(upde3,(0, pad_3_1, 0, pad_3_0), value=0) | ||
86 | de23 = (de2 + upde3pad)/2 | ||
87 | |||
88 | upde23 = self.up12(de23) | ||
89 | pad_23_0 = de1.size()[2] - upde23.size()[2] | ||
90 | pad_23_1 = de1.size()[3] - upde23.size()[3] | ||
91 | upde23pad = F.pad(upde23, (0, pad_23_1, 0, pad_23_0), value=0) | ||
92 | |||
93 | de = (de1 + upde23pad)/2 | ||
83 | 94 | # except Exception as e: | # except Exception as e: |
84 | 95 | # print("EXECEPTION ", e) | # print("EXECEPTION ", e) |
85 | 96 | # print(x.size()) | # print(x.size()) |
File models/playground.py changed (mode: 100644) (index 4aa73cd..b41458f) | |||
... | ... | class M0(nn.Module): | |
25 | 25 | return d | return d |
26 | 26 | ||
27 | 27 | def deconvolutuion(): | def deconvolutuion(): |
28 | transpose = nn.ConvTranspose2d(3, 1, 2, 2) | ||
29 | in_tensor = torch.rand((1, 3, 224, 224)) | ||
28 | transpose = nn.ConvTranspose2d(3, 1, 1, 2, output_padding=1) | ||
29 | in_tensor = torch.rand((1, 3, 40, 40)) | ||
30 | 30 | out = transpose(in_tensor) | out = transpose(in_tensor) |
31 | 31 | print(out.size()) | print(out.size()) |
32 | 32 |