Add the rest of models
Browse files- VFI/gimm-vfi/flowformer_sintel_fp32.safetensors +3 -0
- VFI/gimm-vfi/gimm.yaml +61 -0
- VFI/gimm-vfi/gimmvfi_f_arb.yaml +57 -0
- VFI/gimm-vfi/gimmvfi_f_arb_lpips_fp32.safetensors +3 -0
- VFI/gimm-vfi/gimmvfi_r_arb.yaml +57 -0
- VFI/gimm-vfi/gimmvfi_r_arb_lpips_fp32.safetensors +3 -0
- VFI/gimm-vfi/raft-things_fp32.safetensors +3 -0
- face_restoration/CodeFormer/codeformer.pth +3 -0
- face_restoration/GFPGAN/GFPGANv1.4.pth +3 -0
- face_restoration/GFPGAN/detection_Resnet50_Final.pth +3 -0
- face_restoration/GFPGAN/parsing_parsenet.pth +3 -0
- taehv/taehv.safetensors +3 -0
- taehv/taew2_1.safetensors +3 -0
- upscaling/ESRGAN/4x_NMKD-Siax_200k.safetensors +3 -0
- upscaling/SwinIR/SwinIR.safetensors +3 -0
- yolo/yolov8x-face-lindevs.pt +3 -0
VFI/gimm-vfi/flowformer_sintel_fp32.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:09beaae9b20b924881da175993d0e59535e88beec87363c3a1164184bb914310
|
3 |
+
size 64932100
|
VFI/gimm-vfi/gimm.yaml
ADDED
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
trainer: stage_inr
|
2 |
+
dataset:
|
3 |
+
type: fast_vimeo_flow
|
4 |
+
path: ./data/vimeo90k/vimeo_triplet
|
5 |
+
add_objects: false
|
6 |
+
expansion: false
|
7 |
+
random_t: false
|
8 |
+
aug: true
|
9 |
+
t_scale: 10
|
10 |
+
pair: false
|
11 |
+
|
12 |
+
arch: # needs to add encoder, modulation type
|
13 |
+
type: gimm
|
14 |
+
ema: null
|
15 |
+
|
16 |
+
modulated_layer_idxs: [1]
|
17 |
+
|
18 |
+
coord_range: [-1., 1.]
|
19 |
+
|
20 |
+
hyponet:
|
21 |
+
type: mlp
|
22 |
+
n_layer: 5 # including the output layer
|
23 |
+
hidden_dim: [128] # list, assert len(hidden_dim) in [1, n_layers-1]
|
24 |
+
use_bias: true
|
25 |
+
input_dim: 3
|
26 |
+
output_dim: 2
|
27 |
+
output_bias: 0.5
|
28 |
+
activation:
|
29 |
+
type: siren
|
30 |
+
siren_w0: 1.0
|
31 |
+
initialization:
|
32 |
+
weight_init_type: siren
|
33 |
+
bias_init_type: siren
|
34 |
+
|
35 |
+
loss:
|
36 |
+
type: mse #now unnecessary
|
37 |
+
|
38 |
+
optimizer:
|
39 |
+
type: adam
|
40 |
+
init_lr: 0.0001
|
41 |
+
weight_decay: 0.0
|
42 |
+
betas: [0.9, 0.999] #[0.9, 0.95]
|
43 |
+
ft: false
|
44 |
+
warmup:
|
45 |
+
epoch: 0
|
46 |
+
multiplier: 1
|
47 |
+
buffer_epoch: 0
|
48 |
+
min_lr: 0.0001
|
49 |
+
mode: fix
|
50 |
+
start_from_zero: True
|
51 |
+
max_gn: null
|
52 |
+
|
53 |
+
experiment:
|
54 |
+
amp: True
|
55 |
+
batch_size: 32
|
56 |
+
total_batch_size: 64
|
57 |
+
epochs: 400
|
58 |
+
save_ckpt_freq: 20
|
59 |
+
test_freq: 10
|
60 |
+
test_imlog_freq: 10
|
61 |
+
|
VFI/gimm-vfi/gimmvfi_f_arb.yaml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
trainer: stage_inr
|
2 |
+
dataset:
|
3 |
+
type: vimeo_arb
|
4 |
+
path: ./data/vimeo90k/vimeo_septuplet
|
5 |
+
aug: true
|
6 |
+
|
7 |
+
arch:
|
8 |
+
type: gimmvfi_f
|
9 |
+
ema: true
|
10 |
+
modulated_layer_idxs: [1]
|
11 |
+
|
12 |
+
coord_range: [-1., 1.]
|
13 |
+
|
14 |
+
hyponet:
|
15 |
+
type: mlp
|
16 |
+
n_layer: 5 # including the output layer
|
17 |
+
hidden_dim: [128] # list, assert len(hidden_dim) in [1, n_layers-1]
|
18 |
+
use_bias: true
|
19 |
+
input_dim: 3
|
20 |
+
output_dim: 2
|
21 |
+
output_bias: 0.5
|
22 |
+
activation:
|
23 |
+
type: siren
|
24 |
+
siren_w0: 1.0
|
25 |
+
initialization:
|
26 |
+
weight_init_type: siren
|
27 |
+
bias_init_type: siren
|
28 |
+
|
29 |
+
loss:
|
30 |
+
subsample:
|
31 |
+
type: random
|
32 |
+
ratio: 0.1
|
33 |
+
|
34 |
+
optimizer:
|
35 |
+
type: adamw
|
36 |
+
init_lr: 0.00008
|
37 |
+
weight_decay: 0.00004
|
38 |
+
betas: [0.9, 0.999]
|
39 |
+
ft: true
|
40 |
+
warmup:
|
41 |
+
epoch: 1
|
42 |
+
multiplier: 1
|
43 |
+
buffer_epoch: 0
|
44 |
+
min_lr: 0.000008
|
45 |
+
mode: fix
|
46 |
+
start_from_zero: True
|
47 |
+
max_gn: null
|
48 |
+
|
49 |
+
experiment:
|
50 |
+
amp: True
|
51 |
+
batch_size: 4
|
52 |
+
total_batch_size: 32
|
53 |
+
epochs: 60
|
54 |
+
save_ckpt_freq: 10
|
55 |
+
test_freq: 10
|
56 |
+
test_imlog_freq: 10
|
57 |
+
|
VFI/gimm-vfi/gimmvfi_f_arb_lpips_fp32.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d4da52e7eb636987e0b11c7697bcd35ac4d2172d0227df425e23c1db740fce52
|
3 |
+
size 122632368
|
VFI/gimm-vfi/gimmvfi_r_arb.yaml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
trainer: stage_inr
|
2 |
+
dataset:
|
3 |
+
type: vimeo_arb
|
4 |
+
path: ./data/vimeo90k/vimeo_septuplet
|
5 |
+
aug: true
|
6 |
+
|
7 |
+
arch:
|
8 |
+
type: gimmvfi_r
|
9 |
+
ema: true
|
10 |
+
modulated_layer_idxs: [1]
|
11 |
+
|
12 |
+
coord_range: [-1., 1.]
|
13 |
+
|
14 |
+
hyponet:
|
15 |
+
type: mlp
|
16 |
+
n_layer: 5 # including the output layer
|
17 |
+
hidden_dim: [128] # list, assert len(hidden_dim) in [1, n_layers-1]
|
18 |
+
use_bias: true
|
19 |
+
input_dim: 3
|
20 |
+
output_dim: 2
|
21 |
+
output_bias: 0.5
|
22 |
+
activation:
|
23 |
+
type: siren
|
24 |
+
siren_w0: 1.0
|
25 |
+
initialization:
|
26 |
+
weight_init_type: siren
|
27 |
+
bias_init_type: siren
|
28 |
+
|
29 |
+
loss:
|
30 |
+
subsample:
|
31 |
+
type: random
|
32 |
+
ratio: 0.1
|
33 |
+
|
34 |
+
optimizer:
|
35 |
+
type: adamw
|
36 |
+
init_lr: 0.00008
|
37 |
+
weight_decay: 0.00004
|
38 |
+
betas: [0.9, 0.999]
|
39 |
+
ft: true
|
40 |
+
warmup:
|
41 |
+
epoch: 1
|
42 |
+
multiplier: 1
|
43 |
+
buffer_epoch: 0
|
44 |
+
min_lr: 0.000008
|
45 |
+
mode: fix
|
46 |
+
start_from_zero: True
|
47 |
+
max_gn: null
|
48 |
+
|
49 |
+
experiment:
|
50 |
+
amp: True
|
51 |
+
batch_size: 4
|
52 |
+
total_batch_size: 32
|
53 |
+
epochs: 60
|
54 |
+
save_ckpt_freq: 10
|
55 |
+
test_freq: 10
|
56 |
+
test_imlog_freq: 10
|
57 |
+
|
VFI/gimm-vfi/gimmvfi_r_arb_lpips_fp32.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:897b892e7ee481f2e5c09e59661b645abbd6c9aba9c0c01346bbe31aa569e277
|
3 |
+
size 79204556
|
VFI/gimm-vfi/raft-things_fp32.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba7cb7781f9f67030e2f6acc4151ac8701b633d93d39e8d61991f2b472a0e45a
|
3 |
+
size 21062368
|
face_restoration/CodeFormer/codeformer.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1009e537e0c2a07d4cabce6355f53cb66767cd4b4297ec7a4a64ca4b8a5684b7
|
3 |
+
size 376637898
|
face_restoration/GFPGAN/GFPGANv1.4.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad
|
3 |
+
size 348632874
|
face_restoration/GFPGAN/detection_Resnet50_Final.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6d1de9c2944f2ccddca5f5e010ea5ae64a39845a86311af6fdf30841b0a5a16d
|
3 |
+
size 109497761
|
face_restoration/GFPGAN/parsing_parsenet.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3d558d8d0e42c20224f13cf5a29c79eba2d59913419f945545d8cf7b72920de2
|
3 |
+
size 85331193
|
taehv/taehv.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:032ad4ddc689513287ce02fcc099b5278592526ed1eac0324a7fab94bb690ed9
|
3 |
+
size 22642902
|
taehv/taew2_1.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:79482981ab4db9c2ec909f7aae555831e67efe3e7216752411641b90b9600cdb
|
3 |
+
size 22642902
|
upscaling/ESRGAN/4x_NMKD-Siax_200k.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:839a8ab4d8370a6697e6c07783439c4cea6c5295882745f2e7ddfbb555062286
|
3 |
+
size 66864028
|
upscaling/SwinIR/SwinIR.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b1f58bc2ed44243538356355368ae9cef22d2e13d21485e0206f2cb42f1f3024
|
3 |
+
size 142232532
|
yolo/yolov8x-face-lindevs.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:117c587c79e75e68a83e70200549bf6c035fa45b30e02fb41699639aadcfa0e6
|
3 |
+
size 136757817
|