[misc] Add windows powershell scripts

.gitignore

@@ -20,3 +20,6 @@ data/*.MOV
 data/*.mp4
 data/colmap_text
 data/transforms.json
+lightning_logs
+*.zip
+external

README.md

@@ -58,6 +58,8 @@ Place your video in `data` folder and pass the video path to the script. There a
 ./scripts/train_from_video.sh -v {your_video_name} -s {scale} -f {video_fps}
 ```
 
+Nerf use colmap to get the posture of the camera, so you need to download the colmap from [here](https://github.com/colmap/colmap/releases), and rename the directory to "colmap" and put it in the "external" directory.  
+
 ## [Preview] Mobile Deployment
 
 Using [Taichi AOT](https://docs.taichi-lang.org/docs/tutorial), you can easily deploy a NeRF rendering pipeline on any mobile devices!

data/colmap2nerf.py

@@ -97,7 +97,7 @@ def run_colmap(args):
 
 	# On Windows, if FFmpeg isn't found, try automatically downloading it from the internet
 	if os.name == "nt" and os.system(f"where {colmap_binary} >nul 2>nul") != 0:
-		colmap_glob = os.path.join(ROOT_DIR, "external", "colmap", "*", "COLMAP.bat")
+		colmap_glob = os.path.join(ROOT_DIR, "external", "colmap", "COLMAP.bat")
 		candidates = glob(colmap_glob)
 		if not candidates:
 			print("COLMAP not found. Attempting to download COLMAP from the internet.")

scripts/train_360_v2_garden.ps1

@@ -0,0 +1 @@
+python train.py --root_dir ./360_v2/garden --dataset_name colmap --exp_name garden --downsample 0.25 --no_save_test --num_epochs 20 --scale 16.0 --gui

scripts/train_from_video.ps1

@@ -0,0 +1,20 @@
+#  Put your video in data/ folder and update filename VIDEO_FILE
+#  SCALE choose from 1, 4, 8, 16, 64; 16 is recommended for a real screne
+#  VIDEO_FPS = 2 is suitable for a one minute video 
+set VIDEO_FILE 'video.mp4' 
+set SCALE 16 
+set VIDEO_FPS 2 
+
+echo "video path $VIDEO_FILE"
+echo "scale $SCALE"
+echo "video fps $VIDEO_FPS"
+
+cd "data"
+
+python colmap2nerf.py --video_in $VIDEO_FILE --video_fps $VIDEO_FPS --run_colmap --aabb_scale $SCALE --images images
+
+Move-Item colmap_sparse sparse
+cd ..
+
+
+python train.py --root_dir data --dataset_name colmap --exp_name custom --downsample 0.25  --num_epochs 20 --scale $SCALE  --gui  

scripts/train_nsvf_lego.ps1

@@ -0,0 +1 @@
+python train.py --root_dir "./Synthetic_NeRF/Lego" --exp_name Lego --perf --num_epochs 20 --batch_size 8192 --lr 1e-2 --no_save_test --gui  --ckpt_path=./ckpts/nsvf/Lego/epoch=19-v2.ckpt --val_only