Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Move torchvision.utils.save_image -> torchvision.io.save_image (or introduce this function from scratch)

See original GitHub issue

🚀 The feature

I think the io namespace makes more sense, and it’s where torchvision.io.read_image is located…

Motivation, pitch

N/A

Alternatives

No response

Additional context

No response

Issue Analytics

State:
Created 2 years ago
Comments:9 (3 by maintainers)

Top GitHub Comments

2reactions

ivan-gorincommented, May 9, 2022

I also think save_image should be deprecated, it assumes the input is in [0, 1] range and doesn’t provide any indication that [0, 255] integer representation is not supported. This is also not explained in the documentation.

1reaction

vadimkantorovcommented, Jun 15, 2022

Also, batched/parallelized read_image api could save some simple cases where otherwise a full dataset/dataloader would be needed (if practical enough to implement it).

Top Results From Across the Web

Reading/Writing images and videos - PyTorch

The torchvision.io package provides functions for performing IO operations. They are currently specific to reading and writing video and images.

Getting Started with Albumentation: Deep Learning Image ...

Tutorial on how to use Albumentation library to perform image ... class from torch.utils.data, and crates a TorchVision Dataset Class.

Introduction by Example — pytorch_geometric documentation

We can, e.g., extend it by data.face to save the connectivity of ... PyTorch and torchvision define an example as a tuple of...

Datasets & Dataloaders - PyTorch Tutorials

You can find them here: Image Datasets, Text Datasets, and Audio Datasets ... import torch from torch.utils.data import Dataset from torchvision import ...

Writing CNNs from Scratch in PyTorch - Paperspace Blog

Introduction ; Convolutional Neural Networks; PyTorch; Data Loading ... Use transforms.compose method to reformat images for modeling, # and save to variable ...