Subject | Hash | Author | Date (UTC) |
---|---|---|---|
fix wrong file name | 950ad302d9b0ec6ed04fb4d4a870087dda57281c | Thai Thien | 2020-05-23 16:17:30 |
fix scipy.io | 766d0f3ba48e0c3e0d2562136c5549e20070b805 | Thai Thien | 2020-05-23 15:44:44 |
h2 t3 | 358234af05d250dfccada3b281cd61749ee6c6e4 | Thai Thien | 2020-05-23 15:37:36 |
update experiment_main | e87e1f9e3f0ebe12201c86f61b14a95b9552c199 | Thai Thien | 2020-05-23 14:58:28 |
fix checkAndRecord, add gt count to test data loader | 4290b7bb534ef0f785757b508c8e839ec6b3f9dc | Thai Thien | 2020-05-23 14:38:51 |
log some more | be343dcc3b65f15f30e395eac3ca8271c09d1634 | Thai Thien | 2020-05-21 17:43:04 |
cometml project name | 57ca8f68cbb0b650da74f7117ee60bd25bf0d8d4 | Thai Thien | 2020-05-21 17:39:54 |
train | 1acb89895774528dcec61fad2f2872f2613bff75 | Thai Thien | 2020-05-21 17:34:11 |
train | e67e3fcb8b77db53f990f8582e43c053f69d6c09 | Thai Thien | 2020-05-21 17:31:12 |
fix script | e864e72a9b1fd0e6dfe200b3a9b569016f9894c0 | Thai Thien | 2020-05-21 17:24:38 |
rename | 4f5e84a74bfd4f2a4f60554b9037eb097bafb0df | Thai Thien | 2020-05-21 17:23:08 |
implement the main flow | f31d93a1f3f78575c9ee06bc47b541007789643a | Thai Thien | 2020-05-21 17:16:52 |
script | 4d177e367c6609a0592525ee44b32bcbe43536a6 | Thai Thien | 2020-05-19 16:30:08 |
python code to split train data to train and validate | 0dfb94063b0bdc7aa660b78ab61b9ee5e61a4199 | Thai Thien | 2020-05-18 16:19:21 |
fix dim mismatch | 93ea7669d891301e9c00aadccdea27bb5e138656 | Thai Thien | 2020-05-12 17:19:33 |
train h1 bigtail | ea6391257cd243098cbbb771e705f1f115b845df | Thai Thien | 2020-05-12 16:58:26 |
mse mean | e96c22a36e305681d7fed415a5a949fa0c1791c9 | Thai Thien | 2020-05-10 18:32:02 |
no fix | 7bd97e91de5d7c2d307407287c82e60e893c0c92 | Thai Thien | 2020-05-10 18:22:45 |
no fix | fc20ae6922c2e53f7d37f4228fb921894cd78eab | Thai Thien | 2020-05-10 18:19:59 |
t9 | d8ef865ea602670548e897d8b7ac4c925cc9b393 | Thai Thien | 2020-05-10 18:19:30 |
File | Lines added | Lines deleted |
---|---|---|
data_flow.py | 2 | 2 |
File data_flow.py changed (mode: 100644) (index 11b6709..a520646) | |||
... | ... | def count_gt_annotation_sha(mat_path): | |
30 | 30 | :param mat_path: | :param mat_path: |
31 | 31 | :return: count | :return: count |
32 | 32 | """ | """ |
33 | mat = scipy.io.loadmat(mat_path) | ||
33 | mat = scipy.io.loadmat(mat_path, appendmat=False) | ||
34 | 34 | gt = mat["image_info"][0, 0][0, 0][0] | gt = mat["image_info"][0, 0][0, 0][0] |
35 | 35 | return len(gt) | return len(gt) |
36 | 36 | ||
... | ... | def load_data_shanghaitech_rnd(img_path, train=True): | |
149 | 149 | ||
150 | 150 | if not train: | if not train: |
151 | 151 | # get correct people head count from head annotation | # get correct people head count from head annotation |
152 | mat_path = img_path.replace('.jpg', '.map').replace('images', 'ground-truth').replace('IMG', 'GT_IMG') | ||
152 | mat_path = img_path.replace('.jpg', '.mat').replace('images', 'ground-truth').replace('IMG', 'GT_IMG') | ||
153 | 153 | gt_count = count_gt_annotation_sha(mat_path) | gt_count = count_gt_annotation_sha(mat_path) |
154 | 154 | return img, target1, gt_count | return img, target1, gt_count |
155 | 155 |