Subject | Hash | Author | Date (UTC) |
---|---|---|---|
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 |
add model_context_aware_network.py | 2a36025c001d85afc064c090f4d22987b328977b | Thai Thien | 2020-02-02 03:46:38 |
PACNN (TODO: test this) | 44d5ae7ec57c760fb4f105dd3e3492148a0cc075 | Thai Thien | 2020-02-02 03:40:26 |
add data path | 80134de767d0137a663f343e4606bafc57a1bc1f | Thai Thien | 2020-02-02 03:38:21 |
test if ShanghaiTech datapath is correct | 97ee84944a4393ec3732879b24f614826f8e7798 | Thai Thien | 2020-02-01 03:57:31 |
refactor and test ShanghaiTech datapath | 9542ebc00f257edc38690180b7a4353794be4019 | Thai Thien | 2020-02-01 03:53:49 |
fix the unzip flow | b53c5989935335377eb6a88c942713d3eccc5df7 | Thai Thien | 2020-02-01 03:53:13 |
data_script run seem ok | 67420c08fc1c10a66404d3698994865726a106cd | Thai Thien | 2020-02-01 03:33:18 |
File | Lines added | Lines deleted |
---|---|---|
data_flow.py | 2 | 0 |
train_attn_can_adcrowdnet.py | 1 | 1 |
File data_flow.py changed (mode: 100644) (index ac150b0..cb3c78c) | |||
... | ... | def get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech", | |
363 | 363 | transform=transformer, | transform=transformer, |
364 | 364 | train=False, | train=False, |
365 | 365 | dataset_name=dataset_name), | dataset_name=dataset_name), |
366 | num_workers=4, | ||
366 | 367 | batch_size=1) | batch_size=1) |
367 | 368 | ||
368 | 369 | if test_list is not None: | if test_list is not None: |
... | ... | def get_dataloader(train_list, val_list, test_list, dataset_name="shanghaitech", | |
372 | 373 | transform=transformer, | transform=transformer, |
373 | 374 | train=False, | train=False, |
374 | 375 | dataset_name=dataset_name), | dataset_name=dataset_name), |
376 | num_workers=4, | ||
375 | 377 | batch_size=1) | batch_size=1) |
376 | 378 | else: | else: |
377 | 379 | test_loader = None | test_loader = None |
File train_attn_can_adcrowdnet.py changed (mode: 100644) (index e489e9c..878bdea) | |||
... | ... | if __name__ == "__main__": | |
93 | 93 | score_function=score_function, | score_function=score_function, |
94 | 94 | score_name="mae") | score_name="mae") |
95 | 95 | ||
96 | trainer.add_event_handler(Events.EPOCH_COMPLETED(every=2), save_handler) | ||
96 | trainer.add_event_handler(Events.EPOCH_COMPLETED(every=3), save_handler) | ||
97 | 97 | ||
98 | 98 | trainer.run(train_loader, max_epochs=args.epochs) | trainer.run(train_loader, max_epochs=args.epochs) |