feat: add batch visualization with make_grid and kornia augmentations#106
Open
Saivinay24 wants to merge 3 commits intokornia:masterfrom
Open
feat: add batch visualization with make_grid and kornia augmentations#106Saivinay24 wants to merge 3 commits intokornia:masterfrom
Saivinay24 wants to merge 3 commits intokornia:masterfrom
Conversation
Follow-up to PR kornia#105 as requested by maintainer edgarriba. Changes: - Add kornia to pip install dependencies - New section: Batch Visualization with kornia.image.make_grid - New section: Kornia Augmentations with AugmentationSequential - Before/After comparison grid for augmented batches Augmentations demonstrated: - RandomHorizontalFlip - ColorJitter (brightness, contrast, saturation, hue) - RandomGaussianBlur - RandomRotation
…mpatibility - Execute all cells so outputs are included for PR review - Replace kornia.image.make_grid with torchvision.utils.make_grid (kornia 0.8.2 compat) - Fix n_row -> nrow parameter for torchvision API - Convert conversations column to str() to avoid Arrow nested dict errors - Use image_ds instead of processed_ds in visualization cells to avoid str column type errors
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
for more information, see https://pre-commit.ci
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
Follow-up to #105. This PR enhances the Ray Data + Kornia-RS VLM pipeline tutorial with batch visualization and kornia augmentations, and includes fully executed notebook outputs for inline review.
Changes
torchvision.utils.make_gridto display entire batches as a tiled grid — essential for debugging VLM data pipelines.kornia.augmentation.AugmentationSequentialdemo withRandomHorizontalFlip,ColorJitter,RandomGaussianBlur, andRandomRotation.Compatibility Fixes
kornia.image.make_grid(not available in kornia 0.8.2) withtorchvision.utils.make_gridn_row→nrowparameter for torchvision APIconversationscolumn tostr()to avoid Arrow nested dict conversion errors in Rayimage_ds.select_columns(["pixel_values"])in visualization cells to avoid string column type errorsContent
tutorials/data_loading/ray_vlm_pipeline.ipynbray[data],kornia_rs,kornia, andtorch