Subject | Hash | Author | Date (UTC) |
---|---|---|---|
visualize_shanghaitech_pacnn_with_perspective run without error | 12366a2de2bd60ff4bd36e6132d44e37dedf7462 | Thai Thien | 2020-02-02 04:21:16 |
import ShanghaiTechDataPath in data_util | e81eb56315d44375ff5c0e747d61456601492f8f | Thai Thien | 2020-02-02 04:04:36 |
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 |
add perspective | 642d6fff8c9f31e510fda85a7fb631fb855d8a6d | Thai Thien | 2019-10-06 16:54:44 |
fix padding with p | 86c2fa07822d956a34b3b37e14da485a4249f01b | Thai Thien | 2019-10-06 02:52:58 |
pacnn perspective loss | fb673e38a5f24ae9004fe2b7b93c88991e0c2304 | Thai Thien | 2019-10-06 01:38:28 |
data_flow shanghaitech_pacnn_with_perspective seem working | 91d350a06f358e03223966297d124daee94123d0 | Thai Thien | 2019-10-06 01:31:11 |
multiscale loss and final loss only mode | c65dd0e74ad28503821e5c8651a3b47b4a0c7c64 | Thai Thien | 2019-10-05 15:58:19 |
wip : perspective map | eac63f2671dc5b064753acc4f40bf0f9f216ad2a | Thai Thien | 2019-10-04 16:26:56 |
shell script | f2106e700b6f6174d4dd276f25ec6f3d9ff239bb | thient | 2019-10-04 07:42:51 |
WIP | 42c7c8e1d772fbbda61a4bdf9e329f74e1efb600 | tthien | 2019-10-03 17:52:47 |
add readme | 580cf43d1edddd67b1f6a2c57fdd5cee3dba925c | Thai Thien | 2019-10-02 17:44:49 |
update script, debug | ddb68b95389be1c1d398118677dd227a8bb2b70b | Thai Thien | 2019-10-02 15:52:31 |
add d (output density map) to loss function) | a0c71bf4bf2ab7393d60b06a84db8dfbbfb1a6c2 | tthien | 2019-09-30 16:32:39 |
fix the args, add save interval for model, so we don't save them all | 9fdf9daa2ac4bd12b7b62521d81e520db0debd01 | tthien | 2019-09-30 16:30:00 |
File | Lines added | Lines deleted |
---|---|---|
visualize_data_loader.py | 4 | 1 |
File visualize_data_loader.py changed (mode: 100644) (index 90d8de0..09c9f3b) | |||
... | ... | from data_flow import ListDataset | |
14 | 14 | import pytorch_ssim | import pytorch_ssim |
15 | 15 | ||
16 | 16 | from hard_code_variable import HardCodeVariable | from hard_code_variable import HardCodeVariable |
17 | from data_util import ShanghaiTechDataPath | ||
17 | 18 | from visualize_util import save_img, save_density_map | from visualize_util import save_img, save_density_map |
18 | 19 | ||
19 | 20 | def visualize_ucf_cc_50_pacnn(): | def visualize_ucf_cc_50_pacnn(): |
... | ... | def visualize_ucf_cc_50_pacnn(): | |
51 | 52 | ||
52 | 53 | def visualize_shanghaitech_pacnn_with_perspective(): | def visualize_shanghaitech_pacnn_with_perspective(): |
53 | 54 | HARD_CODE = HardCodeVariable() | HARD_CODE = HardCodeVariable() |
55 | shanghaitech_data = ShanghaiTechDataPath(root=HARD_CODE.SHANGHAITECH_PATH) | ||
56 | shanghaitech_data_part_a_train = shanghaitech_data.get_a().get_train().get() | ||
54 | 57 | saved_folder = "visualize/test_dataloader" | saved_folder = "visualize/test_dataloader" |
55 | 58 | os.makedirs(saved_folder, exist_ok=True) | os.makedirs(saved_folder, exist_ok=True) |
56 | 59 | DATA_PATH = HARD_CODE.SHANGHAITECH_PATH_PART_A | DATA_PATH = HARD_CODE.SHANGHAITECH_PATH_PART_A |
57 | train_list, val_list = get_train_val_list(DATA_PATH, test_size=0.2) | ||
60 | train_list, val_list = get_train_val_list(shanghaitech_data_part_a_train, test_size=0.2) | ||
58 | 61 | test_list = None | test_list = None |
59 | 62 | ||
60 | 63 | # create data loader | # create data loader |