hahattpro / crowd_counting_framework (public) (License: Unspecified) (since 2020-12-23) (hash sha1)
backupfrom https://github.com/ttpro1995/crowd_counting_framework
Clone URLs: https://rocketgit.com/user/hahattpro/crowd_counting_framework ssh://rocketgit@ssh.rocketgit.com/user/hahattpro/crowd_counting_framework git://git.rocketgit.com/user/hahattpro/crowd_counting_framework
debug/a-1 develop feature/add_can_adcrowdnet feature/add_context_aware_network feature/attn_can_adcrowd_simple feature/cache_dataloader feature/compact_cnn feature/compare_with_original_count feature/dataloader_not_divide_8 feature/debug_viz feature/documentation feature/improve_density_gen feature/improve_loss_fn feature/loader_full_and_augment feature/refactor-dataflow feature/refactor_ccnn_basecode feature/save_best feature/simple_cometml feature/train_context_aware_network_on_shtA feature/train_dev_test_split hotfix/fix_visualize_data_loader master meow_exp server-training with_apex
List of commits:
Subject Hash Author Date (UTC)
replace SSIM with kornia SSIM 1f29782fe73751182b696c18567c1de3ab84a6bc Thai Thien 2020-04-30 17:13:09
fix link 3c55cee5cbc9fbaee0859b6ab61529029caee681 Thai Thien 2020-04-30 15:41:10
update documentation feaaa4095d89e99f71f0684eec302d01e421708d Thai Thien 2020-04-30 15:35:38
change gpu c7c6aa29ccee5736c2d4cb290c2ff71c6d897d57 Thai Thien 2020-04-30 12:31:11
reduce batch size 2b2820095211e117c6d02236ef85adb89b70098f Thai Thien 2020-04-30 12:28:32
forget .to(device) a641d5cdf82eb89c9034253785e80dd920c1f5d6 Thai Thien 2020-04-30 12:25:22
reduce batch size of shb keepfull 044f6785fedcf442def0fdb24109413a79274289 Thai Thien 2020-04-30 12:20:21
ccnn v7 with ssim loss 40079e5354eef95e0e89cf23a7f2025ee362e232 Thai Thien 2020-04-30 12:16:19
local 1d4e08754bd493e76cf09076de7c76adf1a30a0a Thai Thien 2020-04-30 12:10:15
mse-ssim loss function 052de7c879bff7690a7cfc1905c8376bf8605c45 Thai Thien 2020-04-30 11:41:24
delete misinformation note ca07533d585def04eec086685f2e72eacc330ddc Thai Thien 2020-04-29 16:59:47
typo 3505077301af5349665f12121862db2512ad450a Thai Thien 2020-04-29 16:59:03
change, adapt, survive 6755c8375302af2146cb63adc967631d53a7b1c8 Thai Thien 2020-04-29 16:48:10
gpu 861ec0d41cea2eec359c6ddfe207e1ed6b583369 Thai Thien 2020-04-28 18:08:11
ccnn_v7_t4 a84f63e64fe8b31fe22c94d383de5ed4e1a27fe4 Thai Thien 2020-04-28 18:07:33
2500 d37516f53d21f3dfc05143ba5ffda80fc5a07825 Thai Thien 2020-04-28 17:56:57
change max epoch 246ae30e6dd2b4a15b7dc70a3dc05592ac1c48f2 Thai Thien 2020-04-28 17:55:23
h1 t3 h1 t4 fbcd13dd240e06a982a1ce48f27cd1d542a26b63 Thai Thien 2020-04-28 17:53:06
h1 96204cb5262500020371637741131d24e3fa91d0 Thai Thien 2020-04-27 17:35:51
typo adfb213c2564bc90b8b69811469534b004808644 Thai Thien 2020-04-27 17:17:58
Commit 1f29782fe73751182b696c18567c1de3ab84a6bc - replace SSIM with kornia SSIM
Author: Thai Thien
Author date (UTC): 2020-04-30 17:13
Committer name: Thai Thien
Committer date (UTC): 2020-04-30 17:13
Parent(s): 3c55cee5cbc9fbaee0859b6ab61529029caee681
Signer:
Signing key:
Signing status: N
Tree: 86e1c59c148cb410568e6a09d3a9dda3ddae1f4e
File Lines added Lines deleted
mse_ssim_loss.py 19 2
train_compact_cnn.py 1 1
File mse_ssim_loss.py changed (mode: 100644) (index babe708..2878cbe)
1 1 import torch import torch
2 2 from torch import nn from torch import nn
3 from pytorch_ssim import SSIM
3 # from pytorch_ssim import SSIM
4
5 from kornia.losses.ssim import SSIM
6
7 # class MseSsimLoss(torch.nn.Module):
8 # """
9 # :deprecated nope, bug, don't use
10 # """
11 # def __init__(self):
12 # super(MseSsimLoss, self).__init__()
13 # self.mse = nn.MSELoss(reduction='sum')
14 # self.ssim = SSIM(window_size=5)
15 #
16 # def forward(self, input, target):
17 # return self.mse(input, target) - self.ssim(input, target)
4 18
5 19
6 20 class MseSsimLoss(torch.nn.Module): class MseSsimLoss(torch.nn.Module):
21 """
22 :deprecated nope, bug, don't use
23 """
7 24 def __init__(self): def __init__(self):
8 25 super(MseSsimLoss, self).__init__() super(MseSsimLoss, self).__init__()
9 26 self.mse = nn.MSELoss(reduction='sum') self.mse = nn.MSELoss(reduction='sum')
10 27 self.ssim = SSIM(window_size=5) self.ssim = SSIM(window_size=5)
11 28
12 29 def forward(self, input, target): def forward(self, input, target):
13 return self.mse(input, target) - self.ssim(input, target)
30 return self.mse(input, target) - self.ssim(input, target)
File train_compact_cnn.py changed (mode: 100644) (index 8f81256..4b570dc)
... ... if __name__ == "__main__":
82 82
83 83 # loss function # loss function
84 84 if args.use_ssim: if args.use_ssim:
85 loss_fn = MseSsimLoss().to(device)
85 loss_fn = MseSsimLoss()
86 86 print("use ssim") print("use ssim")
87 87 else: else:
88 88 loss_fn = nn.MSELoss(reduction='sum').to(device) loss_fn = nn.MSELoss(reduction='sum').to(device)
Hints:
Before first commit, do not forget to setup your git environment:
git config --global user.name "your_name_here"
git config --global user.email "your@email_here"

Clone this repository using HTTP(S):
git clone https://rocketgit.com/user/hahattpro/crowd_counting_framework

Clone this repository using ssh (do not forget to upload a key first):
git clone ssh://rocketgit@ssh.rocketgit.com/user/hahattpro/crowd_counting_framework

Clone this repository using git:
git clone git://git.rocketgit.com/user/hahattpro/crowd_counting_framework

You are allowed to anonymously push to this repository.
This means that your pushed commits will automatically be transformed into a merge request:
... clone the repository ...
... make some changes and some commits ...
git push origin main