wildflow-splat 0.1.3__cp312-cp312-win32.whl → 0.1.4__cp312-cp312-win32.whl

wildflow/__init__.py

@@ -2,6 +2,6 @@
 wildflow - modeling natural ecosystems
 """
 
-__version__ = "0.1.0"
+__version__ = "0.1.4"
 
 __path__ = __import__('pkgutil').extend_path(__path__, __name__)

wildflow/splat/__init__.py

@@ -1,9 +1,24 @@
 """
 wildflow.splat - creating 3D Gaussian Splatting models of coral reefs.
+
+This module provides tools for optimal camera pose partitioning and 
+point cloud processing for 3D Gaussian splatting workflows.
 """
 
+# Core partitioning functionality
+from .patches import patches, BoundingBox
+
+# PLY processing functionality
 from ._core import Config, Patch
 from .split import split_point_cloud
 
-__version__ = "0.1.0"
-__all__ = ["Config", "Patch", "split_point_cloud"]
+__version__ = "0.1.4"
+
+# Public API
+__all__ = [
+    "patches",
+    "BoundingBox", 
+    "Config",
+    "Patch", 
+    "split_point_cloud",
+]

wildflow/splat/patches.py

@@ -0,0 +1,65 @@
+"""
+Split COLMAP camera poses into patches for GPU training.
+
+When you have a large 3D scene with many camera poses (like from COLMAP), 
+you need to split it into smaller patches that can fit on your GPU for 
+3D Gaussian Splatting training. This finds the optimal way to partition 
+your cameras into rectangular patches.
+
+Basic usage:
+    >>> from wildflow.splat import patches
+    >>> # Your camera positions as (x, y) tuples
+    >>> cameras = [(0, 0), (1, 1), (2, 2), (10, 10), (11, 11)]
+    >>> # Split into patches with max 3 cameras each
+    >>> result = patches(cameras, max_cameras=3)
+    >>> print(f"Split into {len(result)} patches")
+    
+Advanced usage:
+    >>> # More control over the splitting
+    >>> result = patches(
+    ...     cameras, 
+    ...     max_cameras=100,    # Max cameras per patch
+    ...     buffer_meters=1.5,  # Safety buffer around patches
+    ...     target_bins=50      # Granularity of splitting
+    ... )
+    
+Loading from COLMAP:
+    >>> import pycolmap
+    >>> model = pycolmap.Reconstruction("path/to/colmap")
+    >>> cameras = [(img.projection_center()[0], img.projection_center()[1]) 
+    ...            for img in model.images.values()]
+    >>> result = patches(cameras)
+"""
+
+from typing import List, Tuple, Sequence
+from ._core import patches as _patches, BoundingBox
+
+# Simple data structures  
+Point = Tuple[float, float]
+
+def patches(camera_positions: Sequence[Point], 
+           max_cameras: int = 700,
+           buffer_meters: float = 1.5, 
+           target_bins: int = 100) -> List[BoundingBox]:
+    """
+    Split camera positions into optimal rectangular patches for GPU training.
+    
+    Args:
+        camera_positions: List of (x, y) camera positions
+        max_cameras: Maximum cameras allowed per patch
+        buffer_meters: Safety buffer around each patch in meters
+        target_bins: Granularity of the splitting algorithm (higher = more precise)
+        
+    Returns:
+        List of BoundingBox patches that cover all cameras optimally
+        
+    Example:
+        >>> cameras = [(0, 0), (1, 1), (2, 2)]
+        >>> result = patches(cameras, max_cameras=10)
+        >>> print(f"Created {len(result)} patches")
+    """
+    # Convert to list of tuples for Rust compatibility
+    camera_list = [(float(pos[0]), float(pos[1])) for pos in camera_positions]
+    
+    # Call the high-performance Rust implementation
+    return _patches(camera_list, max_cameras, buffer_meters, target_bins) 

wildflow_splat-0.1.4.dist-info/METADATA

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: wildflow-splat
+Version: 0.1.4
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Rust
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Multimedia :: Graphics :: 3D Modeling
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: maturin>=1.0 ; extra == 'dev'
+Requires-Dist: pytest>=7.0 ; extra == 'dev'
+Provides-Extra: dev
+License-File: LICENSE
+Summary: Fast PLY point cloud processing for 3D Gaussian splatting workflows
+Keywords: ply,point-cloud,3d-gaussian-splatting,colmap,photogrammetry,computer-vision
+Author-email: Wildflow AI <info@wildflow.ai>
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: repository, https://github.com/wildflowai/wildflow-splat
+Project-URL: documentation, https://docs.wildflow.ai
+
+Super hacky library to work with coral reef splats.
+
+You can swim with a few GoPros around a reef (e.g. [wildflow.ai/protocol](https://wildflow.ai/protocol)) and then turn the footage into 3D models (e.g. [wildflow.ai/demo](https://wildflow.ai/demo)) to track changes over time, run different analysis on top of it, and ultimately see which conservation/restoration methods work best.
+
+This is a bunch of primitives to process the data.
+
+# Usage
+Install with
+```
+pip install wildflow
+```
+So you can play with it from python:
+```py
+from wildflow import splat
+splat.split(...)
+```
+# Workflow
+
+## SfM workflow
+Turns images from cameras 3D point cloud and 
+
+![](/images/wildflow-3dgs-wf.svg)
+
+# Local Development
+
+This library uses Rust extensions built with Maturin. To set up locally:
+
+```bash
+# Create virtual environment
+python3 -m venv venv
+source venv/bin/activate
+
+# Install Rust (if not already installed)
+curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+source "$HOME/.cargo/env"
+
+# Install dependencies and build
+pip install maturin
+pip install -r requirements.txt
+maturin develop
+```
+
+After making changes to Rust code, rebuild with `maturin develop`.
+

wildflow_splat-0.1.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+wildflow/__init__.py,sha256=mY5dWni3TZNOGbUa-7d_VY-Z1GTD7Ebe_OOxwF3Yrls,143
+wildflow/splat/__init__.py,sha256=EPO1QD1-YfTeKxOimF7ACQUexKQZsJHV1fk17cUXW_U,551
+wildflow/splat/_core.cp312-win32.pyd,sha256=AWmpD68XbFEOWWW8AEGOC_BC-QqeyFTusyKNCkzBX6E,629248
+wildflow/splat/patches.py,sha256=OQpqlOpPRRbm_YotxZMDTcnLONmxpjrH7CsNfSekSF0,2515
+wildflow/splat/split.py,sha256=KENk-wXQNEAgZLIokIkmBT3vUpQiSC1NhQAXPNcd1RE,1244
+wildflow_splat-0.1.4.dist-info/METADATA,sha256=nM8wtC5z3OgVGT8RAra8hEDbv8COU_IsQFxZCJ1lCb0,2557
+wildflow_splat-0.1.4.dist-info/WHEEL,sha256=_hRredGV19BUVIQG50bodYTQ5yHFvjyY42Je1tLhKbY,92
+wildflow_splat-0.1.4.dist-info/licenses/LICENSE,sha256=ILWNDv8HAaOBG3VaDh2DsV4XCYCmvNJiSDSimZvyC8U,1108
+wildflow_splat-0.1.4.dist-info/RECORD,,

{wildflow_splat-0.1.3.dist-info → wildflow_splat-0.1.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.6)
+Generator: maturin (1.9.0)
 Root-Is-Purelib: false
 Tag: cp312-cp312-win32

wildflow_splat-0.1.3.dist-info/METADATA

@@ -1,170 +0,0 @@
-Metadata-Version: 2.4
-Name: wildflow-splat
-Version: 0.1.3
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: Intended Audience :: Science/Research
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Rust
-Classifier: Topic :: Scientific/Engineering
-Classifier: Topic :: Multimedia :: Graphics :: 3D Modeling
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Dist: maturin>=1.0 ; extra == 'dev'
-Requires-Dist: pytest>=7.0 ; extra == 'dev'
-Provides-Extra: dev
-License-File: LICENSE
-Summary: Fast PLY point cloud processing for 3D Gaussian splatting workflows
-Keywords: ply,point-cloud,3d-gaussian-splatting,colmap,photogrammetry,computer-vision
-Author-email: Wildflow AI <info@wildflow.ai>
-License: MIT
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
-Project-URL: repository, https://github.com/wildflowai/wildflow-splat
-Project-URL: documentation, https://docs.wildflow.ai
-
-Super hacky library to work with splats.
-
-Usage:
-
-```sh
-git clone https://github.com/wildflowai/splat.git
-cd splat
-pip install -r requirements.txt
-```
-
-# Train Reef Splats Workflow
-
-Someone swim with a few GoPros around a reef and we create 3D model so that you can see corals in the browser and how they change over time (e.g. [wildflow.ai/demo](https://wildflow.ai/demo))
-
-This is a workflow below takes Metashape project as input, trains 3DGS model and deploys it for users to see it.
-
-![](/images/wildflow-3dgs-wf.svg)
-
-# 0. Metashape | 100 years
-
-Metashape project that we recieved for a given coral reef site. Camera positions reconstructed. Everything is scaled correctly. Time-series 3D models are co-registered (aligned in space).
-
-```
-out:
-  metashape/site/time
-                /project_file
-
-retention: R0
-```
-
-# 1. Prep
-
-## 1.a. Colmap | H100 linux | 2 hours
-
-Export camera positions, point cloud, and images (warped) in Colmap format. Keep image size under 2-3MB (sometimes they blow up to 10MB after this).
-
-```
-dep: 0
-out:
-  3dgs/site/time/1_prep/all/images
-                           /sparse/0
-retention: R1 - because metashape and takes time
-```
-
-## 1.b. Colour correction | H100 linux or Windows with GPU | 5 hours
-
-Lightroom or AI colour correction. Keep original warped images.
-
-```
-dep: 1.a
-out:
-  3dgs/site/time/1_prep/all/images
-                           /original_images
-retention: R1 - because it takes time to correct
-```
-
-## 1.c. Train split | H100 linux | 15 min per patch
-
-Split large model into smaller ones for training so that it fits into VRAM. Usually 15x15 meters approx. Use Metashape API to save in COLMAP format these patches.
-
-```
-dep: 0
-out:
-  3dgs/site/time/1_prep/patch/sparse/0
-
-retention: R1 - because of metashape api
-```
-
-## 1.d. Copy images | Any computer would do | 5 min
-
-Copy colour corrected images from the folder with all picturs here, so we have a colmap project for a given patch. Alternatively move colmap artifacts to the folder with all images.
-
-```
-dep: 1.b, 1.c
-out:
-  3dgs/site/time/1_prep/patch/images
-retention: R4 - disposable
-```
-
-# 2. Train | WINDOWS
-
-Train 3DGS for each splat using Postshot. Only works on Windows. Keep both high-resolution and low-resolution models.
-
-```
-dep: 1.d
-
-inp:
-  3dgs/site/time/1_prep/patch
-
-out:
-  3dgs/site/time/2_train/high/site-time-patch.ply
-  3dgs/site/time/2_train/low/site-time-patch.ply
-
-retention: R2 - training takes time and windows
-```
-
-# 3. Cleanup
-
-## 3.a Merge and split | Any computer would do
-
-Split large patches into smaller ones 5x5 cells. Each cell should have 50cm margin (needed for cleaning algo to work well).
-
-```
-dep: 2
-
-out:
-  3dgs/site/time/3_clean/high-raw/site-time-cell.ply
-  3dgs/site/time/3_clean/low-raw/site-time-cell.ply
-
-retention: R4
-```
-
-## 3.b Cleanup | H100 linux
-
-For each cell run cleanup process to remove wacky splats.
-High-res keep as 5x5 cells.
-Low-res merge together into one model.
-
-```
-dep: 3.a
-
-out:
-  3dgs/site/time/3_clean/high/site-time-cell.ply
-  3dgs/site/time/3_clean/low/site-time-cell.ply
-
-retention: R4 - because we deploy this data straightaway if it's good
-```
-
-# 4. Deploy | Any computer would do
-
-Convert clean models into Octree format. Deploy everything to GCS bucket (production) along with credits doc. Now users could fly over coral reefs.
-
-```
-dep: 3.b
-inp: credits.md
-out: gs://wildflow/site-time-hash/...
-
-retention: R1 - because it's user-facing
-```
-

wildflow_splat-0.1.3.dist-info/RECORD

@@ -1,8 +0,0 @@
-wildflow/__init__.py,sha256=c1W_QjZ04hwU8Hhkzk5-17Fj4FjB_ZQIUbFVo7u6-9M,143
-wildflow/splat/__init__.py,sha256=e2sMSGeE4gYJly6EfMcRwI47-yGqh74fCB7rgL-e_uQ,233
-wildflow/splat/_core.cp312-win32.pyd,sha256=9b7CXgW721Q0yCXzm9fUmziKId8JBko2GfLFdOYGvZc,572416
-wildflow/splat/split.py,sha256=KENk-wXQNEAgZLIokIkmBT3vUpQiSC1NhQAXPNcd1RE,1244
-wildflow_splat-0.1.3.dist-info/METADATA,sha256=E0kOni6tHqQkYxUIDa6iILfSs-SioWWnm-zy-hARh3I,4892
-wildflow_splat-0.1.3.dist-info/WHEEL,sha256=iip_4jp1VbEiG4s9Isod9Q3MUK3z1pesoEwCO-vGFn0,92
-wildflow_splat-0.1.3.dist-info/licenses/LICENSE,sha256=ILWNDv8HAaOBG3VaDh2DsV4XCYCmvNJiSDSimZvyC8U,1108
-wildflow_splat-0.1.3.dist-info/RECORD,,