Skip to content

Update leaf segmentation container and inference notebook#35

Open
pradyumnae wants to merge 5 commits into
als-computing:masterfrom
pradyumnae:add-segmentation-container
Open

Update leaf segmentation container and inference notebook#35
pradyumnae wants to merge 5 commits into
als-computing:masterfrom
pradyumnae:add-segmentation-container

Conversation

@pradyumnae
Copy link
Copy Markdown

This PR creates a new Jupyter notebook for running leaf segmentation inference, sets up a modern Docker container based on PyTorch, and resolves GitHub Actions permissions for publishing images. It also cleans up untracked data files and fixes repository issues.

Pradyumna Elavarthi and others added 5 commits March 9, 2026 13:32
- Moved existing Dockerfile/docker-compose.yml/vtk_trial.ipynb into general_model/
- Added new leaf_segmentation/ container with:
  - Dockerfile: Ubuntu + Miniforge + PyTorch (CUDA 12.4) + segmentation-models-pytorch, albumentations, wandb
  - docker-compose.yml: JupyterLab on port 8001
  - Notebook: 12225_UPDATE_TRAINING_fullbigleaf_optionsunetplusplus_batch2.ipynb
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant