List of commits:
Subject Hash Author Date (UTC)
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
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
Commit ddb68b95389be1c1d398118677dd227a8bb2b70b - update script, debug
Author: Thai Thien
Author date (UTC): 2019-10-02 15:52
Committer name: Thai Thien
Committer date (UTC): 2019-10-02 15:52
Parent(s): a0c71bf4bf2ab7393d60b06a84db8dfbbfb1a6c2
Signing key:
Tree: 38d5525d513bf8e34b00ea1e9c757301999c804e
File Lines added Lines deleted
args_util.py 2 1
main_pacnn.py 15 1
train_script/train_pacnn_shanghaitechA.sh 18 2
File args_util.py changed (mode: 100644) (index e7f4e6d..c9c322c)
... ... def real_args_parse():
67 67 parser.add_argument('--model', action="store", default="pacnn") parser.add_argument('--model', action="store", default="pacnn")
68 68
69 69 # args with default value # args with default value
70 parser.add_argument('--load_model', action="store", default="", type=str)
70 71 parser.add_argument('--lr', action="store", default=1e-8, type=float) parser.add_argument('--lr', action="store", default=1e-8, type=float)
71 parser.add_argument('--momentum', action="store", default=0.95, type=float)
72 parser.add_argument('--momentum', action="store", default=0.9, type=float)
72 73 parser.add_argument('--decay', action="store", default=5*1e-3, type=float) parser.add_argument('--decay', action="store", default=5*1e-3, type=float)
73 74 parser.add_argument('--epochs', action="store", default=1, type=int) parser.add_argument('--epochs', action="store", default=1, type=int)
74 75
File main_pacnn.py changed (mode: 100644) (index 9cfb5cc..24d3d1c)
... ... if __name__ == "__main__":
32 32
33 33
34 34 MODEL_SAVE_NAME = args.task_id MODEL_SAVE_NAME = args.task_id
35 MODEL_SAVE_INTERVAL = 10
35 MODEL_SAVE_INTERVAL = 5
36 36 DATA_PATH = args.input DATA_PATH = args.input
37 37 DATASET_NAME = "shanghaitech" DATASET_NAME = "shanghaitech"
38 38 TOTAL_EPOCH = args.epochs TOTAL_EPOCH = args.epochs
 
... ... if __name__ == "__main__":
42 42 experiment.log_parameter("DATA_PATH", DATA_PATH) experiment.log_parameter("DATA_PATH", DATA_PATH)
43 43 experiment.log_parameter("PACNN_PERSPECTIVE_AWARE_MODEL", PACNN_PERSPECTIVE_AWARE_MODEL) experiment.log_parameter("PACNN_PERSPECTIVE_AWARE_MODEL", PACNN_PERSPECTIVE_AWARE_MODEL)
44 44 experiment.log_parameter("train", "train without p") experiment.log_parameter("train", "train without p")
45 experiment.log_parameter("momentum", args.momentum)
46 experiment.log_parameter("lr", args.lr)
45 47
46 48 # create list # create list
47 49 if DATASET_NAME is "shanghaitech": if DATASET_NAME is "shanghaitech":
 
... ... if __name__ == "__main__":
92 94
93 95 current_save_model_name = "" current_save_model_name = ""
94 96 current_epoch = 0 current_epoch = 0
97
98 # load model
99 load_model = args.load_model
100 if len(load_model) > 0:
101 checkpoint = torch.load(load_model)
102 net.load_state_dict(checkpoint['model'])
103 optimizer.load_state_dict(checkpoint['optimizer'])
104 current_epoch = checkpoint['e']
105 print("load ", load_model, " epoch ", str(current_epoch))
106 else:
107 print("new model")
108
95 109 while current_epoch < TOTAL_EPOCH: while current_epoch < TOTAL_EPOCH:
96 110 experiment.log_current_epoch(current_epoch) experiment.log_current_epoch(current_epoch)
97 111 current_epoch += 1 current_epoch += 1
File train_script/train_pacnn_shanghaitechA.sh changed (mode: 100644) (index 448ea1a..b8ab1df)
1 1 #python /home/tt/project/crowd_counting_framework/main_pacnn.py --input /home/tt/project/crowd_counting_framework/data/ShanghaiTech/part_A #python /home/tt/project/crowd_counting_framework/main_pacnn.py --input /home/tt/project/crowd_counting_framework/data/ShanghaiTech/part_A
2 2
3 #python main_pacnn.py \
4 #--input data/ShanghaiTech/part_A \
5 #--epochs 151 \
6 #--task_id train_state1_attemp1
7
8 #python main_pacnn.py \
9 #--input data/ShanghaiTech/part_A \
10 #--load_model saved_model/train_state1_attemp1_10_checkpoint.pth.tar \
11 #--epochs 151 \
12 #--lr 1e-6 \
13 #--task_id train_state1_attemp3
14
15 # trained 30
16
3 17 python main_pacnn.py \ python main_pacnn.py \
4 18 --input data/ShanghaiTech/part_A \ --input data/ShanghaiTech/part_A \
5 --epochs 20 \
6 --task_id train_1
19 --load_model saved_model/train_state1_attemp3_30_checkpoint.pth.tar \
20 --epochs 151 \
21 --lr 1e-7 \
22 --task_id train_state1_attemp4
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