Hugging Face's logo

Spaces:
lupemanzmia
/
video-object-remover
Sleeping

App Files Community
video-object-remover
File size: 1,529 Bytes 
d4b77ac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
 
### General settings
name: FGT_train
use_tb_logger: true
outputdir: /myData/ret/experiments
datadir: /myData
record_iter: 16

### Calling definition
model: model
datasetName_train: train_dataset
network: network

### datasets
datasets:
  train:
    name: youtubevos
    type: video
    mode: train
    dataInfo_config: ./config/data_info.yaml
    use_shuffle: True
    n_workers: 0
    batch_size: 2

  val:
    name: youtubevos
    type: video
    mode: val
    use_shuffle: False
    n_workers: 1
    batch_size: 1
    val_config: ./config/valid_config.yaml

### train settings
train:
  lr: 0.0001
  lr_decay: 0.1
  manual_seed: 10
  BETA1: 0.9
  BETA2: 0.999
  MAX_ITERS: 500000
  UPDATE_INTERVAL: 300000 # 400000 is also OK
  WARMUP: ~
  val_freq: 1  # Set to 1 is for debug, you can enlarge it to 50 in regular training
  TEMPORAL_GAN: ~  # without temporal GAN

### logger
logger:
  PRINT_FREQ: 16
  SAVE_CHECKPOINT_FREQ: 4000   # 100 is for debug consideration

### Data related parameters
flow2rgb: 1
flow_direction: for
num_frames: 5
sample: random
max_val: 0.01

### Model related parameters
res_h: 240
res_w: 432
in_channel: 4
cnum: 64
flow_inChannel: 2
flow_cnum: 64
dist_cnum: 32
frame_hidden: 512
flow_hidden: 256
PASSMASK: 1
num_blocks: 8
kernel_size_w: 7
kernel_size_h: 7
stride_h: 3
stride_w: 3
num_head: 4
conv_type: vanilla
norm: None
use_bias: 1
ape: 1
pos_mode: single
mlp_ratio: 40
drop: 0
init_weights: 1
tw: 2
sw: 8
gd: 4

### Loss weights
L1M: 1
L1V: 1
adv: 0.01

### inference parameters
ref_length: 10