rmuraix commited on
Commit
63f573c
·
1 Parent(s): 05f9d06

feat: add HR-Pro checkpoint (#1)

Browse files

- feat: add HR-Pro checkpoint (dd803e28d0249426d79c9bd2fc7e481b7c141240)

.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ hrpro/ckpt/ub-moji/HR-Pro/stage1/outputs/proposal.json filter=lfs diff=lfs merge=lfs -text
hrpro/ckpt/ub-moji/HR-Pro/stage1/models/model1_seed_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:def033900e630726762d97817530991cce1d0252f65fbbadae13b174eef46cfa
3
+ size 42059302
hrpro/ckpt/ub-moji/HR-Pro/stage1/outputs/config.json ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cfg": "ub-moji",
3
+ "mode": "test",
4
+ "stage": 2,
5
+ "seed": 0,
6
+ "ckpt_path": "./ckpt",
7
+ "backbone": "videoMAE",
8
+ "dataset": "ub-moji",
9
+ "data_path": "./dataset/ub-moji",
10
+ "frames_per_sec": 60,
11
+ "segment_frames_num": 16,
12
+ "feature_dim": 768,
13
+ "tIoU_thresh": [
14
+ 0.1,
15
+ 0.2,
16
+ 0.3,
17
+ 0.4,
18
+ 0.5,
19
+ 0.6,
20
+ 0.7
21
+ ],
22
+ "blocked_videos": [],
23
+ "num_segments": -1,
24
+ "num_workers": 4,
25
+ "num_iters": 5000,
26
+ "test_iter": 50,
27
+ "batch_size": 8,
28
+ "lr": 0.0001,
29
+ "weight_decay": 0.0005,
30
+ "dropout": 0.7,
31
+ "r_act": 8,
32
+ "lambdas": [
33
+ 0,
34
+ 0.5,
35
+ 1,
36
+ 1
37
+ ],
38
+ "RAB_args": {
39
+ "layer_num": 2,
40
+ "drop_out": 0.3,
41
+ "num_heads": 8,
42
+ "dim_feedforward": 128
43
+ },
44
+ "max_epochs": 20,
45
+ "test_epoch": 1,
46
+ "batch_size2": 10,
47
+ "lr2": 5e-05,
48
+ "weight_loss_score": 1,
49
+ "weight_loss_reg": 1,
50
+ "max_proposal": 1000,
51
+ "scale": 2,
52
+ "class_thresh": 0.5,
53
+ "act_thresh_cas": [
54
+ 0.0,
55
+ 0.025,
56
+ 0.05,
57
+ 0.07500000000000001,
58
+ 0.1,
59
+ 0.125,
60
+ 0.15000000000000002,
61
+ 0.17500000000000002,
62
+ 0.2,
63
+ 0.225
64
+ ],
65
+ "act_thresh_agnostic": [
66
+ 0.5,
67
+ 0.525,
68
+ 0.55,
69
+ 0.5750000000000001,
70
+ 0.6000000000000001,
71
+ 0.6250000000000001,
72
+ 0.6500000000000001,
73
+ 0.6750000000000002,
74
+ 0.7000000000000002
75
+ ],
76
+ "_lambda": 0.25,
77
+ "gamma": 0,
78
+ "task_info": "HR-Pro",
79
+ "gpu": "0",
80
+ "root_s1": "./ckpt/ub-moji/HR-Pro/stage1",
81
+ "model_path_s1": "./ckpt/ub-moji/HR-Pro/stage1/models",
82
+ "output_path_s1": "./ckpt/ub-moji/HR-Pro/stage1/outputs",
83
+ "log_path_s1": "./ckpt/ub-moji/HR-Pro/stage1/logs",
84
+ "root_s2": "./ckpt/ub-moji/HR-Pro/stage2",
85
+ "model_path_s2": "./ckpt/ub-moji/HR-Pro/stage2/models",
86
+ "output_path_s2": "./ckpt/ub-moji/HR-Pro/stage2/outputs",
87
+ "log_path_s2": "./ckpt/ub-moji/HR-Pro/stage2/logs",
88
+ "class_name_lst": [
89
+ "a",
90
+ "i",
91
+ "u",
92
+ "e",
93
+ "o",
94
+ "ka",
95
+ "ki",
96
+ "ku",
97
+ "ke",
98
+ "ko",
99
+ "sa",
100
+ "shi",
101
+ "su",
102
+ "se",
103
+ "so",
104
+ "ta",
105
+ "ti",
106
+ "tsu",
107
+ "te",
108
+ "to",
109
+ "na",
110
+ "ni",
111
+ "nu",
112
+ "ne",
113
+ "no",
114
+ "ha",
115
+ "hi",
116
+ "fu",
117
+ "he",
118
+ "ho",
119
+ "ma",
120
+ "mi",
121
+ "mu",
122
+ "me",
123
+ "mo",
124
+ "ya",
125
+ "yu",
126
+ "yo",
127
+ "ra",
128
+ "ri",
129
+ "ru",
130
+ "re",
131
+ "ro",
132
+ "wa",
133
+ "wo",
134
+ "n"
135
+ ],
136
+ "num_class": 46,
137
+ "test_info": {
138
+ "step": [],
139
+ "test_acc": [],
140
+ "loss": [],
141
+ "elapsed": [],
142
+ "now": [],
143
+ "average_mAP[0.1:0.7]": [],
144
+ "average_mAP[0.1:0.5]": [],
145
+ "average_mAP[0.3:0.7]": [],
146
+ "mAP@0.1": [],
147
+ "mAP@0.2": [],
148
+ "mAP@0.3": [],
149
+ "mAP@0.4": [],
150
+ "mAP@0.5": [],
151
+ "mAP@0.6": [],
152
+ "mAP@0.7": []
153
+ }
154
+ }
hrpro/ckpt/ub-moji/HR-Pro/stage1/outputs/proposal.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad0ab043a3351e841ab7ba604c13b9691ca5d32195039a78525ee24fe8394e7d
3
+ size 14743255
hrpro/ckpt/ub-moji/HR-Pro/stage2/models/model2_seed_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20d71d73551a6cb0c3c0524dd0e2719086ba74e46d44724515fb82b3f8d891fa
3
+ size 10634840