[PYTHON] J'ai écrit Gray Scale avec Pytorch

Contexte Quand j'ai essayé de Glay Scale avec Pytorch, il n'y avait pas de fonction, donc je l'ai converti en PIL une fois, Gray Scaled, puis converti à nouveau en Torch Tensor, alors j'ai pensé que c'était une perte de coût de calcul. S'il vous plaît, soutenez-moi autant que l'échelle de gris. .. ..

Je suis désolé s'il y a

Code

À propos, il semble qu'il existe différents types d'échelle de gris, vous pouvez donc utiliser vos paramètres préférés.

    def getGrayImage(self,rgbImg):
        gray = 0.114*rgbImg[:,0,:,:] + 0.587*rgbImg[:,1,:,:] + 0.299*rgbImg[:,2,:,:]
        gray = torch.unsqueeze(gray,1)
        return gray

Conclusion S'il vous plaît laissez-moi savoir s'il y a une fonction GrayScale sur Pytorch!

Références Blague d'image en échelle de gris https://qiita.com/yoya/items/96c36b069e74398796f3

Recommended Posts

J'ai écrit Gray Scale avec Pytorch

J'ai écrit python en japonais

J'ai écrit Fizz Buzz en Python

J'ai écrit la file d'attente en Python

J'ai écrit la pile en Python

Tri sélect écrit en C

J'ai écrit Project Euler 1 en une seule ligne.

J'ai écrit l'aile coulissante dans la création.

J'ai écrit matplotlib

Un mémo que j'ai écrit un tri rapide en Python

J'ai écrit une classe en Python3 et Java

J'ai écrit "Introduction à la vérification des effets" en Python

J'ai écrit un modèle de conception dans l'édition Kotlin Prototype

[PyTorch] J'étais un peu perdu dans torch.max ()

J'ai écrit un analyseur japonais en japonais en utilisant pyparsing.

J'ai écrit un modèle de conception dans l'édition Kotlin Factory

J'ai écrit un modèle de conception dans l'édition Kotlin Builder

J'ai écrit un modèle de conception dans l'édition Kotlin Adapter

J'ai écrit un modèle de conception en kotlin, édité par Iterator

J'ai écrit un modèle de conception dans l'édition de modèle kotlin

J'ai écrit le fonctionnement de base de Seaborn dans Jupyter Lab

J'ai écrit un script de création automatique de répertoire vide en Python

Je l'ai écrit en langage Go pour comprendre le principe SOLID

J'ai écrit un script pour obtenir un site populaire au Japon

J'ai écrit le fonctionnement de base de Numpy dans Jupyter Lab.

[Introduction à Pytorch] Je souhaite générer des phrases dans des articles de presse

J'ai écrit un script qui divise l'image en deux

J'ai écrit GP avec numpy

Implémenter le transfert de style avec Pytorch

Pliez le jeu de données Pytorch en couches

Je comprends Python en japonais!

J'ai créé Word2Vec avec Pytorch

Ce que j'ai appris en Python

J'ai participé à AtCoder (ABC158)

J'ai écrit une fonction pour charger le script d'extension Git en Python

J'ai écrit un script pour extraire les liens de pages Web en Python