Subject | Hash | Author | Date (UTC) |
---|---|---|---|
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 |
meow | 1ad19a22a310992e27a26471feeb37375124d075 | tthien | 2019-09-29 18:25:43 |
fix pacnn perspective map | 453ece3ccb818889ba895bfc4285f7905d33cba5 | Thai Thien | 2019-09-25 17:20:33 |
apex not work so well | da8c0dd57297f972201f31d57e66897177922f48 | Thai Thien | 2019-09-24 17:25:59 |
fix data loader pacnn so it will scale up with correct number of people | 11d55b50d764511f2491291f0208fee0905dec49 | Thai Thien | 2019-09-24 15:40:56 |
add comet ml | a9d4b89ce594f5e241168ccafdcdf0f150ea0ebb | Thai Thien | 2019-09-23 17:07:58 |
fix pacnn avg schema | c2140a96886195782e5689c24aeeb4fe7a2db7ad | Thai Thien | 2019-09-22 17:35:01 |
debug number not divisible by 8 | a568fd7f294a8bd31b3db78437b4b6b51b5b41b9 | Thai Thien | 2019-09-22 04:36:06 |
pacnn | 967074890d14ab0eefc277801860270a468e8f9f | Thai Thien | 2019-09-22 03:54:48 |
wip: pacnn | 2192d7c7b449fecf3868877d9cfbc09bb6f7ae98 | Thai Thien | 2019-09-22 03:44:56 |
wip: pacnn | 37620e5a9bc0f9516ea964ec58d9bdaa1c40ff36 | Thai Thien | 2019-09-22 03:14:42 |
fix training flow | 2b87b1b26c7296b64493fdc49fedb421b249dfa3 | Thai Thien | 2019-09-17 18:00:35 |
dataset script | bc5c052f5f956510ab95ef9a45434fd486c57fae | Thai Thien | 2019-09-16 17:21:13 |
evaluator | ffc5bf8290ae0c469a9a18a2d061cfd1bfeee822 | Thai Thien | 2019-09-14 04:56:35 |
some more test for data loader | 25173578cde7d4e9fe6c6140d1ee01caa4fcfc32 | Thai Thien | 2019-09-14 02:51:58 |
some visualize to debug data loader | e4f52007616acf307bddbde79c0fb4f8c649c785 | Thai Thien | 2019-09-13 17:35:45 |
wip | d7d44cad6774355bdfa45414258763f6c6a0c299 | Thai Thien | 2019-08-31 16:58:16 |
commit all | 6dad7a58f7dbf9fc288ce9dd3e92be538851c2a7 | Thai Thien | 2019-08-29 19:10:44 |
input d1,d2,d3 match | fc2a809241f8b6356d964c63d40cbebd55ca5f6c | Thai Thien | 2019-08-28 17:57:05 |
File | Lines added | Lines deleted |
---|---|---|
main_pacnn.py | 3 | 1 |
File main_pacnn.py changed (mode: 100644) (index c8ba8e9..9cfb5cc) | |||
... | ... | if __name__ == "__main__": | |
117 | 117 | loss_1 = criterion_mse(d1, d1_label) + criterion_ssim(d1, d1_label) | loss_1 = criterion_mse(d1, d1_label) + criterion_ssim(d1, d1_label) |
118 | 118 | loss_2 = criterion_mse(d2, d2_label) + criterion_ssim(d2, d2_label) | loss_2 = criterion_mse(d2, d2_label) + criterion_ssim(d2, d2_label) |
119 | 119 | loss_3 = criterion_mse(d3, d3_label) + criterion_ssim(d3, d3_label) | loss_3 = criterion_mse(d3, d3_label) + criterion_ssim(d3, d3_label) |
120 | loss = loss_1 + loss_2 + loss_3 | ||
120 | loss_d = criterion_mse(d, d1_label) + criterion_ssim(d, d1_label) | ||
121 | loss = loss_d + loss_1 + loss_2 + loss_3 | ||
122 | |||
121 | 123 | if PACNN_PERSPECTIVE_AWARE_MODEL: | if PACNN_PERSPECTIVE_AWARE_MODEL: |
122 | 124 | # TODO: loss for perspective map here | # TODO: loss for perspective map here |
123 | 125 | pass | pass |