Subject | Hash | Author | Date (UTC) |
---|---|---|---|
try to solve x8/8 dimension mismatch | 6b0616197995b3c95041cde43030245f96f1f81e | Thai Thien | 2020-02-27 16:17:12 |
reduce model size, discard middle deform (kernal size 5) | 00fb477cf699a8f2eed8e861f241c077b2046000 | Thai Thien | 2020-02-27 16:11:51 |
shanghaitech same size density map | 25fc2e9395dd67bb6cc273c7fe6cd64c3b154120 | Thai Thien | 2020-02-27 16:03:05 |
fix target1.unsqueeze(0) | bb5c46f121207fd9838ee085314df40513e2ed32 | Thai Thien | 2020-02-27 15:56:07 |
because GPU too small, we crop shanghaitech A | 0f97517dd8de5d3d75cf49af9aa9d20e4595ea3e | Thai Thien | 2020-02-27 15:52:21 |
meow | 3b9576511e597b4894c7c95ca06dc5219ef9c524 | tthien | 2020-02-27 15:44:39 |
fix wrong py name in script, dataset use only 1 worker | 3bd765c9aa79e47049a94547e57deb68c947001c | Thai Thien | 2020-02-27 15:41:44 |
minor fix | 5505b9a8b28df105e11a99d3462323ae9d507102 | Thai Thien | 2020-02-27 15:28:37 |
get ready for short training run with 30 epochs | 66dda0858561897cd5f81e10077459adb39d86dd | Thai Thien | 2020-02-27 15:22:01 |
implement attn_can_adcrowdnet | ffd38664a43d861c20cdc225746b9ce2a00260c7 | Thai Thien | 2020-02-27 15:10:27 |
WIP: add can-adcrowdnet | 5620b83449b31d00a367c8de77e431e19a5ccfb3 | Thai Thien | 2020-02-25 11:31:38 |
add readable timestamp viz | ae1fdb49ddb9ea77659529dceb7fb87c2790c8dc | Thai Thien | 2020-02-24 03:49:57 |
change save name prefix | c53a86f30fb8fd4e8f3a409eb67827d56a43ae5c | Thai Thien | 2020-02-02 10:48:15 |
training flow that work | fb242273e8f696916f9d1ff4bb76b4e5869799ef | Thai Thien | 2020-02-02 10:42:01 |
fix the dataloader for shanghaitech | 5f2aee9f316e6555e6a70c6ad037a4e6b491867b | Thai Thien | 2020-02-02 09:19:50 |
context aware visualize seem ok | 1bdb6ffe77ca4e40ef8f299b2506df2266243db4 | Thai Thien | 2020-02-02 05:07:10 |
visualize eval context aware network seem ok | f3fe45c23dfeab3730624737efabb0b14d23c25b | Thai Thien | 2020-02-02 04:50:34 |
visualize_shanghaitech_pacnn_with_perspective run without error | 12366a2de2bd60ff4bd36e6132d44e37dedf7462 | Thai Thien | 2020-02-02 04:21:16 |
eval context aware network on ShanghaiTechB can run | e8c454d2b6d287c830c1286c9a37884b3cfc615f | Thai Thien | 2020-02-02 04:09:14 |
import ShanghaiTechDataPath in data_util | e81eb56315d44375ff5c0e747d61456601492f8f | Thai Thien | 2020-02-02 04:04:36 |
File | Lines added | Lines deleted |
---|---|---|
models/attn_can_adcrowdnet.py | 4 | 1 |
train_attn_can_adcrowdnet.py | 1 | 1 |
File models/attn_can_adcrowdnet.py changed (mode: 100644) (index 59bb7ff..245aa32) | |||
... | ... | class AttnCanAdcrowdNet(nn.Module): | |
113 | 113 | x = F.relu(self.concat_filter_layer_3(x)) | x = F.relu(self.concat_filter_layer_3(x)) |
114 | 114 | ||
115 | 115 | x = self.output_layer(x) | x = self.output_layer(x) |
116 | x = nn.functional.upsample(x, scale_factor=8, mode='bilinear') / 64.0 | ||
116 | |||
117 | # this cause too much dimension mismatch problem | ||
118 | # so we desampling label instead | ||
119 | # x = nn.functional.upsample(x, scale_factor=8, mode='bilinear') / 64.0 | ||
117 | 120 | return x | return x |
118 | 121 | ||
119 | 122 | def _initialize_weights(self): | def _initialize_weights(self): |
File train_attn_can_adcrowdnet.py changed (mode: 100644) (index 6195c3d..2dbf75f) | |||
... | ... | if __name__ == "__main__": | |
28 | 28 | test_list = None | test_list = None |
29 | 29 | ||
30 | 30 | # create data loader | # create data loader |
31 | train_loader, val_loader, test_loader = get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech_same_size_density_map") | ||
31 | train_loader, val_loader, test_loader = get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech_keepfull") | ||
32 | 32 | ||
33 | 33 | ||
34 | 34 | # model | # model |