[PYTHON] Ich habe Gray Scale mit Pytorch geschrieben

Hintergrund Als ich versuchte, mit Pytorch zu glay Scale, gab es keine Funktion. Ich habe es einmal in PIL konvertiert, Grey Scaled und dann erneut in Torch Tensor konvertiert, sodass ich dachte, es wäre eine Verschwendung von Berechnungskosten. Bitte unterstützen Sie mich genauso wie Gray Scale. .. ..

Es tut mir leid, wenn es gibt

Code

Übrigens scheint es verschiedene Arten von Graustufen zu geben, sodass Sie Ihre bevorzugten Parameter verwenden können.

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

Schlussfolgerung Bitte lassen Sie mich wissen, ob es auf Pytorch eine GrayScale-Funktion gibt!

Referenzen Graustufenbildstreich https://qiita.com/yoya/items/96c36b069e74398796f3

Recommended Posts

Ich habe Gray Scale mit Pytorch geschrieben

Ich habe Python auf Japanisch geschrieben

Ich habe Fizz Buzz in Python geschrieben

Ich habe die Warteschlange in Python geschrieben

Ich habe den Stack in Python geschrieben

Schriftliche Auswahlsortierung in C.

Ich habe Project Euler 1 in einem Liner geschrieben.

Ich habe den Gleitflügel in der Schöpfung geschrieben.

Ich habe matplotlib geschrieben

Ein Memo, das ich schnell in Python geschrieben habe

Ich habe eine Klasse in Python3 und Java geschrieben

Geschrieben "Einführung in die Effektüberprüfung" in Python

Ich habe ein Designmuster in der Kotlin Prototype Edition geschrieben

[PyTorch] Ich war ein wenig verloren in torch.max ()

Ich habe einen japanischen Parser auf Japanisch mit Pyparsing geschrieben.

Ich habe ein Designmuster in der Kotlin Factory Edition geschrieben

Ich habe ein Designmuster in der Kotlin Builder Edition geschrieben

Ich habe ein Designmuster in der Kotlin Adapter Edition geschrieben

Ich habe ein Designmuster in Kotlin geschrieben, das von Iterator bearbeitet wurde

Ich habe ein Designmuster in der Kotlin Template Edition geschrieben

Ich habe die Grundoperation von Seaborn im Jupyter Lab geschrieben

Ich habe ein Skript zur automatischen Erstellung eines leeren Verzeichnisses in Python geschrieben

Ich habe es in der Sprache Go geschrieben, um das SOLID-Prinzip zu verstehen

Ich habe ein Skript geschrieben, um eine beliebte Seite in Japan zu bekommen

Ich habe die Grundoperation von Numpy im Jupyter Lab geschrieben.

[Einführung in Pytorch] Ich möchte Sätze in Nachrichtenartikeln generieren

Ich habe ein Skript geschrieben, das das Bild in zwei Teile teilt

Ich habe GP mit Numpy geschrieben

Implementieren Sie Style Transfer mit Pytorch

Falten Sie den Pytorch-Datensatz in Schichten

Ich verstehe Python auf Japanisch!

Ich habe Word2Vec mit Pytorch gemacht

Was ich in Python gelernt habe

Ich habe an AtCoder (ABC158) teilgenommen.

Ich habe eine Funktion zum Laden des Git-Erweiterungsskripts in Python geschrieben

Ich habe ein Skript geschrieben, um Webseiten-Links in Python zu extrahieren