StableVideo / data /bear /config.json
wchai's picture
init
4c9c42b
{
"results_folder_name": "results_global_rigidity_v2",
"number_of_frames": 70,
"resx": 768,
"resy": 432,
"iters_num": 500000,
"samples_batch": 10000,
"use_optical_flow": true,
"filter_optical_flow": true,
"nerf_skip": true,
"std_coeff": 0.0,
"optical_flow_coeff": 5.0,
"evaluation_resizing_factor": 1,
"evaluate_every": 10000,
"derrivative_amount": 1,
"number_of_channels": 1024,
"rgb_coeff": 5000,
"data_folder": "../../ykasten/neural_video_unwrapping_files/data/pre_processed_davis/bear",
"optimize_only_mask": true,
"jacobian_coeff": 1.0,
"uv_mapping_scale": 0.8,
"train_inv_mapping": false,
"continue_training": false,
"use_simple_mapping_network": true,
"mask_label": -1,
"use_pretrain_mapping": true,
"pretrain_mapping1": true,
"pretrain_mapping2": true,
"continue_training_folder": "",
"erode_mask": false,
"evaluate_all_pixels": false,
"alpha_balancing_factor": 2000.0,
"alpha_flow_factor": 50.0,
"positional_encoding_num_alpha": 5,
"number_of_channels_atlas": 256,
"number_of_layers_atlas": 8,
"number_of_channels_alpha": 256,
"number_of_layers_alpha": 8,
"entropy_scale": 0.0,
"tv_l1_coeff": 100.0,
"L": 0,
"stop_balancing_iteration": 10000,
"background_mask_rcnn_factor": 1.0,
"use_mask_rcnn_bootstrapping": true,
"number_of_channels_mapping1": 256,
"number_of_layers_mapping1": 6,
"number_of_channels_mapping2": 256,
"number_of_layers_mapping2": 4,
"gradient_loss_coeff": 1000,
"use_gradient_loss": true,
"negative_rgb_coeff": 1000.0,
"sparsity_epoch_start": 0,
"sparsity_L2": true,
"sparsity_L1_uniform": false,
"sparsity_size": 10000,
"use_gauss_atlas": false,
"use_gauss_alpha": false,
"freq_num_alpha": 256,
"freq_num_atlas": 512,
"pe_sigma_alpha": 5.0,
"pe_sigma_atlas": 10.0,
"positional_encoding_num_atlas": 10,
"use_weighted_positional_atlas": false,
"pe_weights_N_atlas": 20000,
"use_alpha_rigidity_loss": false,
"use_possitional_encoding_mapping1": false,
"use_gauss_M1": false,
"use_weighted_positional_M1": false,
"use_ppe_M1": false,
"freq_num_M1": 256,
"pe_sigma_M1": 6.0,
"number_of_possitional_encoding_mapping1": 4,
"pe_weights_N_M1": 80000,
"use_possitional_encoding_mapping2": false,
"use_gauss_M2": false,
"use_weighted_positional_M2": false,
"use_ppe_M2": false,
"freq_num_M2": 256,
"pe_sigma_M2": 4.0,
"number_of_possitional_encoding_mapping2": 2,
"pe_weights_N_M2": 80000,
"use_t_features_M1": false,
"t_feat_use_dy_rand_init_M1": false,
"t_feat_use_SDF_rand_init_M1": false,
"t_feat_sigma_M1": 0.01,
"t_num_feat_M1": 512,
"use_t_features_M2": false,
"t_feat_use_dy_rand_init_M2": false,
"t_feat_use_SDF_rand_init_M2": false,
"t_feat_sigma_M2": 0.01,
"t_num_feat_M2": 512,
"train_latent_t_with_different_lr_M1": false,
"train_latent_t_with_different_lr_M2": false,
"t_not_included_atlas": false,
"use_one_optimizer": true,
"pretrain_iter_number": 100,
"save_initialization": true,
"load_initialization": false,
"load_initialization_path": "./results_fixed_weights_and_samples/lucia_05_03_2021__13_35_22_999272/nets_init",
"save_indices_shuffle": false,
"load_indices_shuffle": false,
"load_indices_shuffle_path": "./results_fixed_weights_and_samples/lucia_05_03_2021__13_35_22_999272/jif_shuffle.pt",
"load_checkpoint": false,
"checkpoint_path": "./results_v1_one_optimizer/lucia_04_28_2021__14_32_36_956625/checkpoint",
"use_balanced_sampling": false,
"use_alpha_mat_for_sampling": false,
"balanced_sampling_start": 100000,
"use_sampling_using_optical_flow": false,
"use_sampling_using_optical_flow_timed": false,
"use_sampling_using_optical_flow_start": 12000,
"include_global_rigidity_loss": true,
"global_rigidity_derrivative_amount_fg": 100,
"global_rigidity_derrivative_amount_bg": 100,
"global_jacobian_coeff_fg": 5.0,
"global_jacobian_coeff_bg": 50.0,
"stop_global_rigidity": 5000,
"add_to_folder_name": "_fourth_run",
"use_no_grad_alpha_in_flow_loss": false
}