Subject | Hash | Author | Date (UTC) |
---|---|---|---|
when not train, return img and true count | 93d3b48fc1cd3a5b43ac0e8782c57f2ea00a48bc | Thai Thien | 2020-05-23 16:33:00 |
map->mat | c64a1548b1e9745c109c211f76e8dc2554cf6747 | Thai Thien | 2020-05-23 16:27:48 |
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 |
File | Lines added | Lines deleted |
---|---|---|
crowd_counting_error_metrics.py | 2 | 2 |
data_flow.py | 2 | 2 |
File crowd_counting_error_metrics.py changed (mode: 100644) (index 42efe7b..8eec93f) | |||
... | ... | class CrowdCountingMeanAbsoluteErrorWithCount(Metric): | |
67 | 67 | self._num_examples = 0 | self._num_examples = 0 |
68 | 68 | ||
69 | 69 | def update(self, output): | def update(self, output): |
70 | y_pred, y, true_count = output | ||
70 | y_pred, true_count = output | ||
71 | 71 | pred_count = torch.sum(y_pred) | pred_count = torch.sum(y_pred) |
72 | 72 | # true_count = torch.sum(y) | # true_count = torch.sum(y) |
73 | 73 | absolute_errors = torch.abs(pred_count - true_count) | absolute_errors = torch.abs(pred_count - true_count) |
... | ... | class CrowdCountingMeanSquaredErrorWithCount(Metric): | |
92 | 92 | self._num_examples = 0 | self._num_examples = 0 |
93 | 93 | ||
94 | 94 | def update(self, output): | def update(self, output): |
95 | y_pred, y, true_count = output | ||
95 | y_pred, true_count = output | ||
96 | 96 | pred_count = torch.sum(y_pred) | pred_count = torch.sum(y_pred) |
97 | 97 | # true_count = torch.sum(y) | # true_count = torch.sum(y) |
98 | 98 | squared_errors = torch.pow(pred_count - true_count, 2) | squared_errors = torch.pow(pred_count - true_count, 2) |
File data_flow.py changed (mode: 100644) (index ee67cd1..96d7ec7) | |||
... | ... | def load_data_shanghaitech_rnd(img_path, train=True): | |
151 | 151 | # get correct people head count from head annotation | # get correct people head count from head annotation |
152 | 152 | mat_path = img_path.replace('.jpg', '.mat').replace('images', 'ground-truth').replace('IMG', 'GT_IMG') | 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 | return img, target1, gt_count | ||
154 | return img, gt_count | ||
155 | 155 | ||
156 | 156 | return img, target1 | return img, target1 |
157 | 157 | ||
... | ... | def load_data_shanghaitech_20p(img_path, train=True): | |
280 | 280 | # get correct people head count from head annotation | # get correct people head count from head annotation |
281 | 281 | mat_path = img_path.replace('.jpg', '.mat').replace('images', 'ground-truth').replace('IMG', 'GT_IMG') | mat_path = img_path.replace('.jpg', '.mat').replace('images', 'ground-truth').replace('IMG', 'GT_IMG') |
282 | 282 | gt_count = count_gt_annotation_sha(mat_path) | gt_count = count_gt_annotation_sha(mat_path) |
283 | return img, target1, gt_count | ||
283 | return img, gt_count | ||
284 | 284 | ||
285 | 285 | return img, target1 | return img, target1 |
286 | 286 |