List of commits:
Subject Hash Author Date (UTC)
fix pacnn perspective map 453ece3ccb818889ba895bfc4285f7905d33cba5 Thai Thien 2019-09-25 17:20:33
apex not work so well da8c0dd57297f972201f31d57e66897177922f48 Thai Thien 2019-09-24 17:25:59
fix data loader pacnn so it will scale up with correct number of people 11d55b50d764511f2491291f0208fee0905dec49 Thai Thien 2019-09-24 15:40:56
add comet ml a9d4b89ce594f5e241168ccafdcdf0f150ea0ebb Thai Thien 2019-09-23 17:07:58
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
Commit 453ece3ccb818889ba895bfc4285f7905d33cba5 - fix pacnn perspective map
Author: Thai Thien
Author date (UTC): 2019-09-25 17:20
Committer name: Thai Thien
Committer date (UTC): 2019-09-25 17:20
Parent(s): da8c0dd57297f972201f31d57e66897177922f48
Signing key:
Tree: 63d1ca896becc4f6870c964f896ad56e9a807890
File Lines added Lines deleted
main_pacnn.py 4 3
models/pacnn.py 21 2
models/test_PACNNWithPerspectiveMap.py 8 0
File main_pacnn.py changed (mode: 100644) (index 1078b73..43b6c3d)
... ... if __name__ == "__main__":
28 28 MODEL_SAVE_NAME = "dev5" MODEL_SAVE_NAME = "dev5"
29 29 # Add the following code anywhere in your machine learning file # Add the following code anywhere in your machine learning file
30 30 experiment = Experiment(api_key="S3mM1eMq6NumMxk2QJAXASkUM", experiment = Experiment(api_key="S3mM1eMq6NumMxk2QJAXASkUM",
31 project_name="pacnn-dev", workspace="ttpro1995")
31 project_name="pacnn-dev2", workspace="ttpro1995")
32 32 experiment.set_name(MODEL_SAVE_NAME) experiment.set_name(MODEL_SAVE_NAME)
33
33 experiment.log_parameter("PACNN_PERSPECTIVE_AWARE_MODEL", True)
34 experiment.log_parameter("desc", "test pacnn with perspective aware")
34 35 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
35 36
36 37 # device = "cpu" # device = "cpu"
 
... ... if __name__ == "__main__":
39 40 print(args) print(args)
40 41 DATA_PATH = args.input DATA_PATH = args.input
41 42 DATASET_NAME = "shanghaitech" DATASET_NAME = "shanghaitech"
42 PACNN_PERSPECTIVE_AWARE_MODEL = False
43 PACNN_PERSPECTIVE_AWARE_MODEL = True
43 44
44 45
45 46
File models/pacnn.py changed (mode: 100644) (index 46f1e5d..0336b5c)
... ... class PACNNWithPerspectiveMap(nn.Module):
72 72 if self.perspective_aware_mode: if self.perspective_aware_mode:
73 73 pespective_w_s = self.perspective_11(self.perspective_net(x)) pespective_w_s = self.perspective_11(self.perspective_net(x))
74 74 pespective_w = self.up_perspective(pespective_w_s) pespective_w = self.up_perspective(pespective_w_s)
75
76
75 77 # TODO: code more here # TODO: code more here
76 de23 = pespective_w_s * de2 + (1 - pespective_w_s)*(de2 + self.up23(de3))
77 de = pespective_w * de1 + (1 - pespective_w)*(de1 + self.up12(de23))
78 upde3 = self.up23(de3)
79 pad_3_0 = de2.size()[2] - upde3.size()[2]
80 pad_3_1 = de2.size()[3] - upde3.size()[3]
81 upde3pad = F.pad(upde3, (0, pad_3_1, 0, pad_3_0), value=0)
82 de23_a = pespective_w_s * de2
83 de23_b = (1 - pespective_w_s)*(de2 + upde3pad)
84 de23 = de23_a + de23_b
85
86 upde23 = self.up12(de23)
87 pad_23_0 = de1.size()[2] - upde23.size()[2]
88 pad_23_1 = de1.size()[3] - upde23.size()[3]
89 upde23pad = F.pad(upde23, (0, pad_23_1, 0, pad_23_0), value=0)
90
91 pad_perspective_0 = de1.size()[2] - pespective_w.size()[2]
92 pad_perspective_1 = de1.size()[3] - pespective_w.size()[3]
93 pespective_w_pad = F.pad(pespective_w, (0, pad_perspective_1, 0, pad_perspective_0), value=0)
94 de_a = pespective_w_pad * de1
95 de_b = (1 - pespective_w_pad)*(de1 + upde23pad)
96 de = de_a + de_b
78 97 else: else:
79 98 #try: #try:
80 99 pespective_w_s = None pespective_w_s = None
File models/test_PACNNWithPerspectiveMap.py changed (mode: 100644) (index af8340f..a6b4678)
... ... class TestPACNNWithPerspectiveMap(TestCase):
42 42 image = torch.rand(1, 3, 224, 224) image = torch.rand(1, 3, 224, 224)
43 43 _, _, _, _, _, density_map = net(image) _, _, _, _, _, density_map = net(image)
44 44 print(density_map.size()) print(density_map.size())
45
46 def test_perspective_aware_schema_pacnn_pad(self):
47 net = PACNNWithPerspectiveMap(perspective_aware_mode=True)
48 # image
49 # batch size, channel, h, w
50 image = torch.rand(1, 3, 330, 512)
51 _, _, _, _, _, density_map = net(image)
52 print(density_map.size())
Hints:
Before first commit, do not forget to setup your git environment:
git config --global user.name "your_name_here"
git config --global user.email "your@email_here"

Clone this repository using HTTP(S):
git clone https://rocketgit.com/user/hahattpro/crowd_counting_framework

Clone this repository using ssh (do not forget to upload a key first):
git clone ssh://rocketgit@ssh.rocketgit.com/user/hahattpro/crowd_counting_framework

Clone this repository using git:
git clone git://git.rocketgit.com/user/hahattpro/crowd_counting_framework

You are allowed to anonymously push to this repository.
This means that your pushed commits will automatically be transformed into a merge request:
... clone the repository ...
... make some changes and some commits ...
git push origin main