Skip to content
Snippets Groups Projects
Commit a6960c82 authored by mhby1g21's avatar mhby1g21
Browse files

changed edgenet360 to edgenet-360

parent 49e35ccc
No related branches found
No related tags found
No related merge requests found
...@@ -6,7 +6,7 @@ $imageNameOrId = (Get-Content $configPath | Where-Object { $_ -match "imageId=" ...@@ -6,7 +6,7 @@ $imageNameOrId = (Get-Content $configPath | Where-Object { $_ -match "imageId="
$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path $scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
# Calculate paths # Calculate paths
$rootDir = Split-Path -Parent (Split-Path -Parent $scriptDir) $rootDir = Split-Path -Parent (Split-Path -Parent $scriptDir)
$outputDir = Join-Path $rootDir "edgenet360\Data\Input" $outputDir = Join-Path $rootDir "edgenet-360\Data\Input"
# Docker container name or ID # Docker container name or ID
$containerNameOrId = docker ps --filter "ancestor=$imageNameOrId" --format "{{.ID}}" $containerNameOrId = docker ps --filter "ancestor=$imageNameOrId" --format "{{.ID}}"
......
...@@ -20,8 +20,8 @@ def shift_image_selection(): ...@@ -20,8 +20,8 @@ def shift_image_selection():
pass pass
def copy_intermediary_outputs(): def copy_intermediary_outputs():
source_folder = os.path.join(ROOT_DIR, "edgenet360", "Data", "Input") source_folder = os.path.join(ROOT_DIR, "edgenet-360", "Data", "Input")
destination_folder = os.path.join(ROOT_DIR, "edgenet360", "Output") destination_folder = os.path.join(ROOT_DIR, "edgenet-360", "Output")
files_to_copy = ["depth_e.png", "enhanced_depth_e.png", "material.png", "rgb.png"] files_to_copy = ["depth_e.png", "enhanced_depth_e.png", "material.png", "rgb.png"]
for file_name in files_to_copy: for file_name in files_to_copy:
......
...@@ -20,13 +20,13 @@ for /f "tokens=1,2 delims==" %%a in (config.ini) do ( ...@@ -20,13 +20,13 @@ for /f "tokens=1,2 delims==" %%a in (config.ini) do (
set scriptDir=%~dp0 set scriptDir=%~dp0
set workingDir=%scriptDir%.. set workingDir=%scriptDir%..
set monoDepthDir=%workingDir%\scripts\360monodepthexecution set monoDepthDir=%workingDir%\scripts\360monodepthexecution
set outputDir=%workingDir%\edgenet360\Output set outputDir=%workingDir%\edgenet-360\Output
set materialRecogDir=%workingDir%\Dynamic-Backward-Attention-Transformer set materialRecogDir=%workingDir%\Dynamic-Backward-Attention-Transformer
set edgeNetDir=%workingDir%\edgenet360 set edgeNetDir=%workingDir%\edgenet-360
:: File paths :: File paths
set checkpointFile=%materialRecogDir%\checkpoints\dpglt_mode95\accuracy\epoch=126-valid_acc_epoch=0.87.ckpt set checkpointFile=%materialRecogDir%\checkpoints\dpglt_mode95\accuracy\epoch=126-valid_acc_epoch=0.87.ckpt
set shiftedImage=%scriptDir%shifted_t.png set shiftedImage=%scriptDir%\shifted_t.png
set monoDepthImage=%monoDepthDir%\rgb.jpg set monoDepthImage=%monoDepthDir%\rgb.jpg
:: Ensure the input file exists :: Ensure the input file exists
...@@ -82,7 +82,7 @@ popd ...@@ -82,7 +82,7 @@ popd
echo Running material recognition... echo Running material recognition...
pushd "%materialRecogDir%" pushd "%materialRecogDir%"
call "%condaDir%\condabin\activate.bat" %materialEnv% call "%condaDir%\condabin\activate.bat" %materialEnv%
python train_sota.py --data-root "./datasets" --batch-size 1 --tag dpglt --gpus 1 --num-nodes 1 --epochs 200 --mode 95 --seed 42 --test "%checkpointFile%" --infer "%materialRecogDir%/split_output/" python train_sota.py --data-root "./datasets" --batch-size 1 --tag dpglt --gpus 1 --num-nodes 1 --epochs 200 --mode 95 --seed 42 --test "%checkpointFile%" --infer "%materialRecogDir%/cubemap_faces/"
if errorlevel 1 ( if errorlevel 1 (
echo Error: Material recognition failed. Please check the output above for more details. echo Error: Material recognition failed. Please check the output above for more details.
call "%condaDir%\condabin\deactivate.bat" call "%condaDir%\condabin\deactivate.bat"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment