Commit dcba5844 authored by Nianchen Deng's avatar Nianchen Deng
Browse files

update run_spherical_view_syn.py

parent 2a1d7973
...@@ -87,44 +87,35 @@ ...@@ -87,44 +87,35 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 2, "execution_count": 5,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
"name": "stdout", "name": "stdout",
"output_type": "stream", "output_type": "stream",
"text": [ "text": [
"Change working directory to /home/dengnc/dvs/data/__new/lobby_all\n", "Change working directory to /home/dengnc/dvs/data/__new/barbershop_all\n",
"Load net from fovea@snerffast4-rgb_e6_fc512x4_d2.00-50.00_s64_~p.pth ...\n", "Load net from fovea@snerffast4-rgb_e6_fc512x4_d1.20-6.00_s64_~p.pth ...\n",
"Load net from periph@snerffast4-rgb_e6_fc256x4_d2.00-50.00_s64_~p.pth ...\n" "Load net from periph@snerffast4-rgb_e6_fc256x4_d1.20-6.00_s64_~p.pth ...\n"
] ]
} }
], ],
"source": [ "source": [
"scene = 'lobby'\n", "scene = 'barbershop'\n",
"os.chdir(f'{rootdir}/data/__new/{scenes[scene]}')\n", "os.chdir(f'{rootdir}/data/__new/{scenes[scene]}')\n",
"print('Change working directory to ', os.getcwd())\n", "print('Change working directory to ', os.getcwd())\n",
"\n", "\n",
"fovea_net = load_net(find_file('fovea'))\n", "fovea_net = load_net(find_file('fovea'))\n",
"periph_net = load_net(find_file('periph'))\n", "periph_net = load_net(find_file('periph'))\n",
"renderer = FoveatedNeuralRenderer(fov_list, res_list, nn.ModuleList([fovea_net, periph_net, periph_net]),\n", "renderer = FoveatedNeuralRenderer(fov_list, res_list, nn.ModuleList([fovea_net, periph_net, periph_net]),\n",
" res_full, device=device.default())\n" " res_full, using_mask=False, device=device.default())\n"
] ]
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 15, "execution_count": null,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [],
{
"name": "stderr",
"output_type": "stream",
"text": [
"/home/dengnc/miniconda3/lib/python3.8/site-packages/torch/nn/functional.py:3828: UserWarning: Default grid_sample and affine_grid behavior has changed to align_corners=False since 1.3.0. Please specify align_corners=True if the old behavior is desired. See the documentation of grid_sample for details.\n",
" warnings.warn(\n"
]
}
],
"source": [ "source": [
"params = {\n", "params = {\n",
" 'classroom': [\n", " 'classroom': [\n",
...@@ -137,9 +128,10 @@ ...@@ -137,9 +128,10 @@
" [0, 0, 0, 0, 10, 200, -50]\n", " [0, 0, 0, 0, 10, 200, -50]\n",
" ],\n", " ],\n",
" 'barbershop': [\n", " 'barbershop': [\n",
" [0, 0, 0, 20, 0, -300, 50],\n", " [0, 0, 0, 0, 0, 0, 0],\n",
" [0, 0, 0, -140, -30, 150, -250],\n", " #[0, 0, 0, 20, 0, -300, 50],\n",
" [0, 0, 0, -60, -30, 75, -125],\n", " #[0, 0, 0, -140, -30, 150, -250],\n",
" #[0, 0, 0, -60, -30, 75, -125],\n",
" ],\n", " ],\n",
" 'lobby': [\n", " 'lobby': [\n",
" #[0, 0, 0, 0, 0, 75, 0],\n", " #[0, 0, 0, 0, 0, 75, 0],\n",
......
...@@ -406,7 +406,8 @@ def test(): ...@@ -406,7 +406,8 @@ def test():
out[key][global_idx] = ret[key] out[key][global_idx] = ret[key]
if args.output_flags['perf']: if args.output_flags['perf']:
perf_times[i] = perf.checkpoint() perf_times[i] = perf.checkpoint()
progress_bar(i, n, 'Inferring...') if not args.log_redirect:
progress_bar(i, n, 'Inferring...')
i += 1 i += 1
global_offset += n_rays global_offset += n_rays
......
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