Upload 16 files

CSTVR/archived/Contin/inverter/config.yml

@@ -0,0 +1,130 @@
+# GENERATE TIME: Thu Jun 20 11:02:09 2024
+# CMD:
+# basicsr/train_inv.py --local_rank=0 -opt options/Contin/surrogate_contin.yml --launcher pytorch
+
+# general settings for image SR training
+name: INDModel_contin_adobe
+model_type: INDModel
+scale: 2
+num_gpu: 7
+manual_seed: 10
+
+# dist:
+#   find_unused_parameters: true
+# dataset and data loader settings
+datasets:
+  train: 
+    S: -1
+    T: -1
+    task: INDModel_contin
+    name: Adobe_Rate
+    type: Adobe_Rate
+    image_dir:  'path/to/your/dir'
+    base_p: 'path/to/your/dir'
+    base_num: 121
+    gt_size: 160
+    data_augmentation: true
+    load_mode: 'train'
+    seq_skip: 50
+    base_num: 121
+    # data loader
+    use_shuffle: true
+    num_worker_per_gpu: 1
+    batch_size_per_gpu: 1
+    dataset_enlarge_ratio: 1
+    prefetch_mode: ~
+
+  val:
+    S: -1
+    T: -1
+    task: INDModel_contin
+    name: Adobe_Rate
+    type: Adobe_Rate
+    image_dir:  'path/to/your/dir'
+    base_p: 'path/to/your/dir'
+    base_num: 121
+    gt_size: 160
+    data_augmentation: false
+    load_mode: 'test'
+    seq_skip: 50
+    base_num: 121
+    # data loader
+    use_shuffle: false
+down_type:
+  S: -1
+  T: -1
+
+# network structures
+network_g:
+  type: IND_inv3D
+  opt:
+    pretrained_down: 
+    block_num: 2
+    down_opt:
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+      quant_type: round
+    up_opt:
+      in_channels: 3
+      three_D_res_num: 24
+      mid_channels: 128
+      fuse_channels: 96
+   
+
+
+# path
+path:
+  pretrain_network:
+  strict_load_g: true
+  resume_state: 
+
+# training settings
+train:
+  optim_g:
+    type: Adam
+    #lr: !!float 2e-4
+    lr: !!float 2e-4
+    weight_decay: 0
+    betas: [0.9, 0.99]
+
+  scheduler:
+    type: MultiStepLR
+    milestones: [35000, 70000, 105000,140000]
+    gamma: 0.5
+
+  total_iter: 140000
+  warmup_iter: -1  # no warm up
+
+  # losses
+  pixel_opt:
+    type: CharbonnierLoss
+    loss_weight: 1.0
+    reduction: mean
+# validation settings
+val:
+  val_freq:  999999999999999999999
+  save_img: false
+
+  metrics:
+    psnr: # metric name, can be arbitrary
+      type: calculate_psnr
+      crop_border: 0
+      test_y_channel: true
+
+
+
+# logging settings
+logger:
+  print_freq: 100
+  save_checkpoint_freq:  3000
+  use_tb_logger: true
+  wandb:
+    project: ~
+    resume_id: ~
+
+# dist training settings
+dist_params:
+  backend: nccl
+  port: 29500

CSTVR/archived/Contin/inverter/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a845c0cddcaa4889928032570223539e64011dbb22a966b1b951123d2b28d5d
+size 16194234

CSTVR/archived/Contin/rescaler/config.yml

@@ -0,0 +1,62 @@
+
+# dataset and data loader settings
+datasets:
+  train: 
+    S: -1
+    T: -1
+    task: INDModel_contin
+    name: Adobe_Rate
+    type: Adobe_Rate
+    image_dir:  '/home/zhangyuantong/dataset/adobe/train/'
+    base_p: '/home/zhangyuantong/dataset/adobe/train/'
+    base_num: 121
+    gt_size: 160
+    data_augmentation: true
+    load_mode: 'train'
+    seq_skip: 50
+    base_num: 121
+    # data loader
+    use_shuffle: true
+    num_worker_per_gpu: 1
+    batch_size_per_gpu: 1
+    dataset_enlarge_ratio: 1
+    prefetch_mode: ~
+
+  val:
+    S: -1
+    T: -1
+    task: INDModel_contin
+    name: Adobe_Rate
+    type: Adobe_Rate
+    image_dir:  '/home/zhangyuantong/dataset/adobe/test/'
+    base_p: '/home/zhangyuantong/dataset/adobe/test/'
+    base_num: 121
+    gt_size: 160
+    data_augmentation: false
+    load_mode: 'test'
+    seq_skip: 50
+    base_num: 121
+    # data loader
+    use_shuffle: false
+    
+down_type:
+  S: -1
+  T: -1 # -1表示连续
+  # S: 1
+  # T: 2
+# network structures
+network_g:
+  type: RescalerNet
+  opt:
+    quan_type: round
+    control_rate: false
+    down_opt:
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+    up_opt:
+      in_channels: 3
+      three_D_res_num: 24
+      mid_channels: 128
+      fuse_channels: 96

CSTVR/archived/Contin/rescaler/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:876de685a6000971f3ade59ad470f65cd764183894b94584ffa3a3a2bfe7051a
+size 47069348

CSTVR/archived/Tx1_Sx4_vimeo/inverter/config.yml

@@ -0,0 +1,18 @@
+
+down_type:
+  S: 4
+  T: 1
+# network structures
+network_g:
+  type: IND_inv3D
+  opt:
+    block_num: 2
+    down_opt:
+      quant_type: round
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+   
+
+

CSTVR/archived/Tx1_Sx4_vimeo/inverter/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1c008e1a16b40aaa6a7830cc1592a044d62515193d701846390cf9dc07b9fd2
+size 16194298

CSTVR/archived/Tx1_Sx4_vimeo/rescaler/config.yml

@@ -0,0 +1,39 @@
+datasets:
+  val:
+    name: Vimeo_SepTuplet_val
+    type: Vimeo_SepTuplet
+    gt_size: 160
+    T_index: [0,1,2,3,4,5,6]
+    S_scale: 4
+    image_dir: path/to/your/dir
+    file_list: sep_testlist.txt
+    load_mode: 'test'
+    data_augmentation: false
+down_type:
+  S: 4
+  T: 1
+# network structures
+network_g:
+  type: Rescaler_MixNet
+  opt:
+    quan_type: round
+    control_rate: false
+    down_opt:
+      quan_type: round
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+    up_opt:
+      use_shortcut: true
+      use_shuffle: false
+      in_channels: 3
+      fuse_channels: 64
+      input_dim: 96
+      dim: 150
+      out_dim: 64
+      depths: [6, 6, 6, 6]
+      num_heads: [6,6,6,6]
+      window_size: [3,7,7]
+      three_D_res_num: 4
+      fuse_channels: 96

CSTVR/archived/Tx1_Sx4_vimeo/rescaler/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7d5fa31024d412d822b002c1333b97d4410d582a5d2b44a129fdccfb050fbbf
+size 73801198

CSTVR/archived/Tx2_Sx1_vimeo/inverter/config.yml

@@ -0,0 +1,15 @@
+
+down_type:
+  S: 1
+  T: 2
+# network structures
+network_g:
+  type: IND_inv3D
+  opt:
+    block_num: 4
+    down_opt:
+      quant_type: round
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]

CSTVR/archived/Tx2_Sx1_vimeo/inverter/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97e9cdbd5149fd2e0776d0971c55a17d4f21d87a18dbbf1c5a3ae6cd57cab5ba
+size 26151970

CSTVR/archived/Tx2_Sx1_vimeo/rescaler/config.yml

@@ -0,0 +1,34 @@
+
+# dataset and data loader settings
+datasets:
+  val:
+    name: Vimeo_SepTuplet_val
+    type: Vimeo_SepTuplet
+    gt_size: 128
+    T_index: [0,2,4,6]
+    S_scale: 4
+    image_dir: /home/zhangyuantong/dataset/vimeo_septuplet/sequences/
+    file_list: sep_testlist.txt
+    load_mode: 'test'
+    data_augmentation: false
+down_type:
+  S: 1
+  T: 2
+
+# network structures
+network_g:
+  type: RescalerNet
+  opt:
+    control_rate: false
+    quan_type: round
+    down_opt:  
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+    up_opt:
+      in_channels: 3
+      three_D_res_num: 24
+      mid_channels: 128
+      fuse_channels: 96
+     

CSTVR/archived/Tx2_Sx1_vimeo/rescaler/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:04d6db6f363c80e8b26be18b945fd340ff6ac71a338289a3b95f10741d1d0368
+size 47069348

CSTVR/archived/Tx2_Sx4_vimeo/inverter/config.yml

@@ -0,0 +1,10 @@
+network_g:
+  type: IND_inv3D
+  opt:
+    block_num: 4
+    down_opt:
+      quant_type: round
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]

CSTVR/archived/Tx2_Sx4_vimeo/inverter/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c4f57c54b9dc16ca50c070874c1c584bcf09f64edc8942a00b6962a8dd765549
+size 26151970

CSTVR/archived/Tx2_Sx4_vimeo/rescaler/config.yml

@@ -0,0 +1,41 @@
+
+# dataset and data loader settings
+datasets:
+  val:
+    name: Vimeo_SepTuplet_val
+    type: Vimeo_SepTuplet
+    gt_size: 128
+    T_index: [0,2,4,6]
+    S_scale: 4
+    image_dir: /home/zhangyuantong/dataset/vimeo_septuplet/sequences/
+    file_list: sep_testlist.txt
+    load_mode: 'test'
+    data_augmentation: false
+down_type:
+  S: 4
+  T: 2
+# network structures
+network_g:
+  type: Rescaler_MixNet
+  opt: 
+    control_rate: false
+    quan_type: round
+    down_opt:
+      in_channels: 3
+      mid_channels: 64
+      three_D_res_num: 6
+      kernel_size: [3,3,3]
+    up_opt:
+      use_shuffle: false
+      use_shortcut: true
+      in_channels: 3
+      fuse_channels: 64
+      input_dim: 96
+      dim: 150
+      out_dim: 64
+      depths: [6, 6, 6, 6]
+      num_heads: [6,6,6,6]
+      window_size: [3,7,7]
+      three_D_res_num: 4
+      fuse_channels: 96
+

CSTVR/archived/Tx2_Sx4_vimeo/rescaler/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d82a1b2fead86ad32687464165506dac7a2e38f5e48e52042345ee042381606
+size 73799283