pytorch / pytorch/vision

torchvision.utils.make_grid pad_value attribute to accept RGB value

Open
#4,526 5 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

🚀 Feature

change torchvision.utils.make_grid pad_value attribute to accept an RGB value (in whatever format, either hex string, or 3-tuple)

Motivation

This would allow to pad image grids with colors other than just grayscale

Pitch

I want to be able to pass an RGB value to pad_value of torchvision.utils.make_grid

Alternatives

changing the padded value manually, that's quite tedious. Or just not using this function.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start at torchvision.utils.make_grid and inspect how pad_value is currently applied when padding image grids. Define the accepted RGB representation from the issue, then verify that colored padding is produced without breaking existing grayscale behavior.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
computer-vision
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
42/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.