Hugging Face's logo

Spaces:
venite
/
sat3density
Runtime error

App Files Community
sat3density
File size: 2,768 Bytes 
f670afc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
_parent_: options/base.yaml

gpu_ids: '0'

## config for wandb
project: 'sat2pano'
Group: 'craft_feature'


model: craft_feature                           # model/craft_feature
arch:   
  gen:                                         ## config for generator
    netG: imaginaire.generators.craft_2stage_add_style  
    weight_norm_type: spectral
    activation_norm_type: instance
    padding_mode: reflect
    transform_mode: volum_rendering
    feature_model:
    style_inject: histo                        # use histogram to inject illumination, chose list [histo, perspective]
    cat_PE: 
    cat_opa: true
    cat_depth: true
    depth_arch:                                # Density Net
      name: depth
      num_filters: 32                          
      num_downsamples: 4                       
      num_res_blocks: 6                        
      output_nc: 64
    render_arch:                               # Render Net
      name: render
      num_filters: 64                          
      num_downsamples: 4                       
      num_res_blocks: 9                        
      output_nc: 3
    style_enc_cfg:                              # style injection
      input_image_channels: 3
      num_filters: 256
      kernel_size: 3
      style_dims: 128
      interm_style_dims: 256
      hidden_channel: 256
      weight_norm_type: spectral
  dis:                                          # discriminator
    netD: imaginaire.discriminators.multires_patch_pano
    num_filters: 64
    max_num_filters: 512
    num_discriminators: 3
    num_layers: 3
    weight_norm_type: spectral
    activation_norm_type: instance



data:                                            # data options
  dataset: CVACT_Shi                           # dataset name
  root: ./dataset/CVACT/ 
  sat_size: [256,256]              
  pano_size: [512, 128]
  sample_number: 100                           # points per ray
  max_height: 8                                # pre-defined density space in height axis
  sky_mask: true                               
  histo_mode: rgb
  # val: 
  #   sub: 500        


optim:   
  lr_gen: 0.00005                                
  lr_dis: 0.00005
  gan_mode: non_saturated                         #'hinge', 'least_square',  'non_saturated', 'wasserstein'
  loss_weight:
    L1: 1   
    L2: 10
    GaussianKL: 0.1
    feature_matching: 10.0
    Perceptual: 10
    sky_inner: 1 
    GAN: 1                                              

  lr_policy:
    iteration_mode: False                         # iteration or epoch
    type: step
    step_size: 45                         
    gamma: 0.1

  ground_prior: true 

######## for test, if only style, will random choice one style for save dir
only_style: 
only_img:
save_dir: