Как конвертировать несколько картинок в папку с цветовой картой ITK? - PullRequest
Новая
       53

Как конвертировать несколько картинок в папку с цветовой картой ITK?

0 голосов
/ 22 декабря 2018

Каждый раз, когда я запускаю свою программу в терминале, она печатает: thumb0496.jpg не конвертируется {} не конвертируется

Что бы я ни делал, это никогда не работает ... Я новичок в Python и установилэто через Анаконду вместе с OpenCV, Pip и ITK.Я занимаюсь этим всего 4 дня и застрял.Python - мой родной язык.Почему мой код не работает?

Если этот код выглядит похожим, то это так.Я должен был попробовать объединить некоторые элементы.К сожалению, я не могу найти пост снова.Код был хуже, но я (как-то) исправил его.Именно эту (новую) часть я не могу исправить самостоятельно!

import cv2
import sys
import itk
import os,glob

from os import listdir,makedirs
from os.path import isfile,join
path = '/Users/admin/Desktop/ff'
dstpath = '/Users/admin/Desktop/test'

PixelType = itk.UC
Dimension = 2

ImageType = itk.Image[PixelType, Dimension]

RGBPixelType = itk.RGBPixel[PixelType]
RGBImageType = itk.Image[RGBPixelType, Dimension]

ColormapType = itk.CustomColormapFunction[PixelType, RGBPixelType]
colormap = ColormapType.New()

ColormapFilterType = itk.ScalarToRGBColormapImageFilter[ImageType,RGBImageType]
colormapFilter1 = ColormapFilterType.New()

colormapFilter1.SetInput(reader.GetOutput())
colormapFilter1.SetColormap(colormap)

WriterType = itk.ImageFileWriter[RGBImageType]
writer = WriterType.New()
writer.SetFileName(dstpath)
writer.SetInput(colormapFilter1.GetOutput())

try:
    makedirs(dstpath)
except:
    print ("Directory already exist, images will be written in same folder")
files = [f for f in listdir(path) if isfile(join(path,f))]
for image in files:
    try:
        reader = ReaderType(os.path.join(path,image))
        map = ColormapFilterType(reader, PixelType, RGBImageType, ImageType)
        dstPath = join(dstpath,image)
        cv2.imwrite(dstPath,map)
    except:
        print ("{} is not converted".format(image))
for fil in glob.glob("*.jpg"):
    try:
        img = ReaderType(os.path.join(path,fil))
        map_imag = ColormapType(img, PixelType, RGBImageType,ImageType)
        cv2.imwrite(os.path.join(dstpath,fil),map_image)
    except:
        print('{} is not converted')

1 Ответ

0 голосов
/ 24 декабря 2018

Почему бы вам не начать с рабочего примера и постепенно изменить его в соответствии со своими потребностями?Примеры можно найти в кратком руководстве и в этом блоге post .

...