feat: experimental all-intra mode coding

Author: JasonLSC

examples/benchmarks/mpeg/video_anchor_all_intra_bench.sh

@@ -0,0 +1,47 @@
+#!/bin/bash
+
+# Define the list of GPU IDs to use
+GPU_IDS=(4 5 6 7)  # You can modify this list, e.g., GPU_IDS=(0 2 5 7)
+
+# Function to run a single experiment
+run_experiment() {
+    local gpu_id=$1
+    local rp_id=$2
+    
+    echo "Starting experiment rp${rp_id} on GPU ${gpu_id}"
+    
+    CUDA_VISIBLE_DEVICES=${gpu_id} python compress_ply_sequence.py x265_compression_rp${rp_id} \
+        --data_factor 1 \
+        --ply_dir /work/Users/lisicheng/Dataset/GSC_splats/m71763_bartender_stable/track \
+        --data_dir /work/Users/lisicheng/Dataset/GSC_splats/m71763_bartender_stable/colmap_data \
+        --result_dir results/mpeg150/video_anchor_all_intra/rp${rp_id} \
+        --frame_num 16 \
+        --lpips_net vgg \
+        --no-normalize_world_space \
+        --scene_type GSC \
+        --test_view_id 9 11 \
+        --compression_cfg.use_sort \
+        --compression_cfg.use_all_intra
+    
+    echo "Experiment rp${rp_id} started on GPU ${gpu_id}"
+}
+
+# Check if the number of GPUs is sufficient
+if [ ${#GPU_IDS[@]} -lt 4 ]; then
+    echo "Warning: Number of GPUs is less than the number of experiments, some experiments will be skipped"
+fi
+
+# Launch experiments in parallel
+for i in {0..3}; do
+    if [ $i -lt ${#GPU_IDS[@]} ]; then
+        run_experiment ${GPU_IDS[$i]} $i &
+        echo "Launched experiment rp${i} on GPU ${GPU_IDS[$i]} in background"
+    else
+        echo "Skipping experiment rp${i} due to insufficient GPUs"
+    fi
+done
+
+# Wait for all background processes to complete
+wait
+
+echo "All experiments completed"

examples/benchmarks/mpeg/video_anchor_all_intra_debug.sh

@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# Define the list of GPU IDs to use
+GPU_IDS=(4 5 6 7)  # You can modify this list, e.g., GPU_IDS=(0 2 5 7)
+
+# Function to run a single experiment
+run_experiment() {
+    local gpu_id=$1
+    local rp_id=$2
+    
+    echo "Starting experiment rp${rp_id} on GPU ${gpu_id}"
+    
+    CUDA_VISIBLE_DEVICES=${gpu_id} python compress_ply_sequence.py x265_compression_rp${rp_id} \
+        --data_factor 1 \
+        --ply_dir /work/Users/lisicheng/Dataset/GSC_splats/m71763_bartender_stable/track \
+        --data_dir /work/Users/lisicheng/Dataset/GSC_splats/m71763_bartender_stable/colmap_data \
+        --result_dir results/mpeg150/video_anchor_all_intra_debug/rp${rp_id} \
+        --frame_num 16 \
+        --lpips_net vgg \
+        --no-normalize_world_space \
+        --scene_type GSC \
+        --test_view_id 9 11 \
+        --compression_cfg.use_sort \
+        --compression_cfg.use_all_intra \
+        --compression_cfg.debug
+    
+    echo "Experiment rp${rp_id} started on GPU ${gpu_id}"
+}
+
+# Check if the number of GPUs is sufficient
+if [ ${#GPU_IDS[@]} -lt 4 ]; then
+    echo "Warning: Number of GPUs is less than the number of experiments, some experiments will be skipped"
+fi
+
+# Launch experiments in parallel
+for i in {0..3}; do
+    if [ $i -lt ${#GPU_IDS[@]} ]; then
+        run_experiment ${GPU_IDS[$i]} $i &
+        echo "Launched experiment rp${i} on GPU ${GPU_IDS[$i]} in background"
+    else
+        echo "Skipping experiment rp${i} due to insufficient GPUs"
+    fi
+done
+
+# Wait for all background processes to complete
+wait
+
+echo "All experiments completed"

examples/compress_ply_sequence.py

@@ -114,6 +114,8 @@ class VideoCompressionConfig(CompressionConfig):
     n_clusters: int = 32768
     # Maps attribute names to their codec functions
     attribute_codec_registry: Optional[AttributeCodecs] = field(default_factory=lambda: AttributeCodecs())
+    # Enable All Intra coding mode
+    use_all_intra: bool = False
     # Enable debug mode
     debug: bool = False
 
@@ -122,13 +124,13 @@ def to_dict(self) -> Dict[str, Any]:
         Convert the CompressionConfig instance to a dictionary.
         If attribute_codec_registry is not None, it will be converted to a dictionary using its to_dict method.
         """
-        result = {
-            "use_sort": self.use_sort,
-            "verbose": self.verbose,
-            "qp": self.qp,
-            "n_clusters": self.n_clusters,
-            "debug": self.debug,
-        }
+        # Get only attributes defined in VideoCompressionConfig
+        video_compression_attrs = [
+            "use_sort", "verbose", "qp", "n_clusters", 
+            "use_all_intra", "debug"
+        ]
+        
+        result = {attr: getattr(self, attr) for attr in video_compression_attrs}
 
         # handle attribute_codec_registry
         if self.attribute_codec_registry is not None: