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
|