Subject | Hash | Author | Date (UTC) |
---|---|---|---|
fix args dataset name | 55aa10b342aa2921639ddb214c91a1ece401d400 | Thai Thien | 2020-02-29 02:04:00 |
fix deprecate | d16b3ed34285df19d950d4f0472867020922cb91 | Thai Thien | 2020-02-29 01:56:38 |
fix dataset_name | 16a6b9ccc89d647c239778662d1371f4e43c864a | Thai Thien | 2020-02-29 01:40:50 |
fix my_args_parse | c0f526e0affc64cc00134a84c8c7275e91297b70 | Thai Thien | 2020-02-29 01:36:50 |
minor fix | 8f3909b1a4c62e67bc2a8c2f218fc0697ff7cef3 | Thai Thien | 2020-02-29 01:35:32 |
train script | 96afb84e92320e5389a302964b6a5d68186f6435 | Thai Thien | 2020-02-29 01:32:14 |
done | 12f18f211de9ae25ee3ed25812cef1c7a361b31d | Thai Thien | 2020-02-29 01:30:11 |
fix error: argument --decay: invalid float value: '5*1e-4' | a144c83e69e4363311fa74f1c091d1a3f834ad8a | Thai Thien | 2020-02-28 17:16:41 |
change optim to Adam, lr and decay same with content aware network | 0b0879b22e9d202015e67a0cc8c87ddcafb3c2d2 | Thai Thien | 2020-02-28 17:15:05 |
train with higher learning rate | 3fa7e328dc4b7b4570fd32908b3df5fab8283919 | Thai Thien | 2020-02-28 16:57:16 |
train 25 epoch only | 1cfb58b5eae299f49e3b64289c3a12a8f35cee45 | Thai Thien | 2020-02-27 18:35:58 |
we freeze vgg pretrain layer | 459cefbfbf53949cb9636ca002cd4f78e964bb83 | Thai Thien | 2020-02-27 18:24:17 |
fix error when save | 86a8f467d47a67da0accc963e8baacbaea07ab14 | Thai Thien | 2020-02-27 17:55:46 |
we discard deform 7 and keep deform 5 instead | 91ff78e4a1aef3677b15447b631c2927c9e62a31 | Thai Thien | 2020-02-27 16:22:47 |
reduce image by crop | 530516ff40b744cac79756c97faaed50c8b65cfb | Thai Thien | 2020-02-27 16:19:11 |
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 |
File | Lines added | Lines deleted |
---|---|---|
train_attn_can_adcrowdnet_simple.py | 5 | 2 |
File train_attn_can_adcrowdnet_simple.py changed (mode: 100644) (index 04edd18..2c852b2) | |||
... | ... | if __name__ == "__main__": | |
21 | 21 | DATA_PATH = args.input | DATA_PATH = args.input |
22 | 22 | TRAIN_PATH = os.path.join(DATA_PATH, "train_data") | TRAIN_PATH = os.path.join(DATA_PATH, "train_data") |
23 | 23 | TEST_PATH = os.path.join(DATA_PATH, "test_data") | TEST_PATH = os.path.join(DATA_PATH, "test_data") |
24 | |||
24 | dataset_name = args.datasetname | ||
25 | if (dataset_name=="shanghaitech"): | ||
26 | print("so true") | ||
25 | 27 | ||
26 | 28 | # create list | # create list |
27 | 29 | train_list, val_list = get_train_val_list(TRAIN_PATH) | train_list, val_list = get_train_val_list(TRAIN_PATH) |
28 | 30 | test_list = None | test_list = None |
29 | 31 | ||
30 | 32 | # create data loader | # create data loader |
31 | train_loader, val_loader, test_loader = get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech") | ||
33 | |||
34 | train_loader, val_loader, test_loader = get_dataloader(train_list, val_list, test_list, dataset_name=dataset_name) | ||
32 | 35 | ||
33 | 36 | ||
34 | 37 | # model | # model |