Skip to content

GitLab

Commit 6294701e authored by Nianchen Deng's avatar Nianchen Deng
Browse files

sync

parent 2824f796
Show whitespace changes
Inline Side-by-side
configs/snerfadvx4_voxels.json 0 → 100644
View file @ 6294701e
{
"model": "SNeRFX",
"args": {
"color": "rgb",
"encode_x": 10,
"encode_d": 4,
"core": "nerfadv",
"core_params": {
"density_net": {
"nf": 128,
"n_layers": 4
},
"color_net": {
"nf": 128,
"n_layers": 3
},
"specular_net": {
"nf": 128,
"n_layers": 1
},
"appearance": "combined"
},
"n_featdim": 0,
"space": "_nets/train_t0.3/snerfadv_voxels+ls2/checkpoint_50.tar",
"n_samples": 256,
"multi_nets": 4
},
"train": {
"max_epochs": 50,
"prune_epochs": [5],
"split_epochs": [10],
"density_regularization_weight": 1e-4,
"density_regularization_scale": 1e4
}
}
\ No newline at end of file
configs/snerfadvx_voxels_x4.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFAdvanceX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"density_net": {
"nf": 128,
"n_layers": 4,
"act": "relu",
"skips": []
},
"color_net": {
"nf": 128,
"n_layers": 3,
"act": "relu",
"skips": []
},
"specular_net": {
"nf": 128,
"n_layers": 1,
"act": "relu"
},
"n_featdim": 0,
"space": "_nets/train_t0.3/snerfadv_voxels+ls2/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"appearance": "combined",
"density_color_connection": true,
"density_regularization_weight": 1e-4,
"density_regularization_scale": 1e4,
"multi_nets": 4
}
}
\ No newline at end of file
configs/snerfadvx_voxels_x8.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFAdvanceX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"density_net": {
"nf": 128,
"n_layers": 4,
"act": "relu",
"skips": []
},
"color_net": {
"nf": 128,
"n_layers": 3,
"act": "relu",
"skips": []
},
"specular_net": {
"nf": 128,
"n_layers": 1,
"act": "relu"
},
"n_featdim": 0,
"space": "_nets/hr_t1.0s/snerfadv_voxels+ls2/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"appearance": "combined",
"density_color_connection": true,
"density_regularization_weight": 1e-4,
"density_regularization_scale": 1e4,
"multi_nets": 8
}
}
\ No newline at end of file
configs/snerfx_voxels_128x4_x4.json → configs/snerfx4_voxels.json
View file @ 6294701e
......@@ -2,20 +2,22 @@
"model": "SNeRFX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"encode_x": 10,
"encode_d": 4,
"core_params": {
"nf": 128,
"n_layers": 4,
"activation": "relu",
"skips": []
"n_layers": 4
},
"n_featdim": 0,
"space": "nets/train1/snerf_voxels/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"raymarching_tolerance": 0,
"raymarching_chunk_size": -1,
"multi_nets": 4
},
"train": {
"max_epochs": 50,
"prune_epochs": [5],
"split_epochs": [10],
"density_regularization_weight": 1e-4,
"density_regularization_scale": 1e4
}
}
\ No newline at end of file
configs/snerfx_voxels_128x4_x8.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"nf": 128,
"n_layers": 4,
"activation": "relu",
"skips": []
},
"n_featdim": 0,
"space": "nets/train_t0.3/snerf_voxels/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"raymarching_tolerance": 0,
"raymarching_chunk_size": -1,
"multi_nets": 8
}
}
\ No newline at end of file
configs/snerfx_voxels_128x8_x4.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"nf": 128,
"n_layers": 8,
"activation": "relu",
"skips": [4]
},
"n_featdim": 0,
"space": "nets/train_t0.3/snerf_voxels/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"raymarching_tolerance": 0,
"raymarching_chunk_size": -1,
"multi_nets": 4
}
}
\ No newline at end of file
configs/snerfx_voxels_256x4_x4.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"nf": 256,
"n_layers": 4,
"activation": "relu",
"skips": []
},
"n_featdim": 0,
"space": "nets/train1/snerf_voxels/checkpoint_50.tar",
"n_samples": 256,
"perturb_sample": true,
"raymarching_tolerance": 0,
"raymarching_chunk_size": -1,
"multi_nets": 4
}
}
\ No newline at end of file
configs/snerfx_voxels_256x4_x4_balance.json deleted 100644 → 0
View file @ 2824f796
{
"model": "SNeRFX",
"args": {
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"nf": 256,
"n_layers": 4,
"activation": "relu",
"skips": []
},
"n_featdim": 0,
"space": "voxels",
"steps": [4, 16, 8],
"n_samples": 16,
"perturb_sample": true,
"raymarching_tolerance": 0,
"raymarching_chunk_size": -1,
"multi_nets": 4
}
}
\ No newline at end of file
configs/snerf_voxels+ls+f32.json → configs/svnerf_voxels.json
View file @ 6294701e
{
"model": "SNeRF",
"model": "VNeRF",
"args": {
"spherical": true,
"color": "rgb",
"n_pot_encode": 10,
"n_dir_encode": 4,
"fc_params": {
"encode_x": 10,
"encode_d": 4,
"core_params": {
"nf": 256,
"n_layers": 8,
"activation": "relu",
"skips": [ 4 ]
},
"n_featdim": 32,
"space": "voxels",
"steps": [4, 16, 8],
"n_samples": 16,
"perturb_sample": true,
"n_samples": 32
},
"train": {
"max_epochs": 50,
"prune_epochs": [5],
"split_epochs": [10],
"density_regularization_weight": 1e-4,
"density_regularization_scale": 1e4
}
......
construct_dataset.sh 0 → 100755
View file @ 6294701e
#!/bin/bash
scene=$1
basename=pano_t0.8
echo Copy from blender output folder...
cp -r "../blender_models/$scene/output/hr_${basename}" "data/$scene/hr_${basename}_raw"
cp "../blender_models/$scene/output/hr_${basename}.json" "data/$scene/"
cd "data/$scene" || exit 1
cp "hr_${basename}.json" "mr_${basename}.json"
sed -i "s/\"x\": [0-9]\+/\"x\": 1800/g" "mr_${basename}.json"
sed -i "s/\"y\": [0-9]\+/\"y\": 900/g" "mr_${basename}.json"
cp "hr_${basename}.json" "lr_${basename}.json"
sed -i "s/\"x\": [0-9]\+/\"x\": 900/g" "lr_${basename}.json"
sed -i "s/\"y\": [0-9]\+/\"y\": 450/g" "lr_${basename}.json"
echo Scale raw images...
python ../../tools/image_scale.py "hr_${basename}_raw" "mr_${basename}_raw" --width 1800 --height 900
python ../../tools/image_scale.py "mr_${basename}_raw" "lr_${basename}_raw" --width 900 --height 450
echo Process pano images...
python ../../tools/pano_process.py "hr_${basename}_raw" -o "hr_${basename}"
python ../../tools/pano_process.py "mr_${basename}_raw" -o "mr_${basename}"
python ../../tools/pano_process.py "lr_${basename}_raw" -o "lr_${basename}"
echo Clean...
rm -r "hr_${basename}_raw"
rm -r "mr_${basename}_raw"
rm -r "lr_${basename}_raw"
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment