Не удается записать образ SimpleElastix (Python)

Я пытаюсь запустить следующий пример как часть библиотеки SimpleElastix:

import SimpleITK as sitk

elastixImageFilter = sitk.ElastixImageFilter()
elastixImageFilter.SetFixedImage(sitk.ReadImage('1.jpg', sitk.sitkFloat32))
elastixImageFilter.SetMovingImage(sitk.ReadImage('2.jpg', sitk.sitkFloat32))
elastixImageFilter.SetParameterMap(sitk.GetDefaultParameterMap('rigid'))
elastixImageFilter.Execute()
sitk.WriteImage(elastixImageFilter.GetResultImage())

Когда я пытаюсь запустить приведенный выше код, я получаю следующую ошибку (я показываю часть вывода):

Traceback (most recent call last):
  File "rigid_transform.py", line 8, in <module>
    sitk.WriteImage(elastixImageFilter.GetResultImage())
  File "/usr/local/lib/python2.7/dist-packages/SimpleElastix-1.0.1rc1.dev331+gd756f-py2.7-linux-x86_64.egg/SimpleITK/SimpleITK.py", line 8015, in WriteImage
    return _SimpleITK.WriteImage(*args)
NotImplementedError: Wrong number or type of arguments for overloaded function 'WriteImage'.
  Possible C/C++ prototypes are:
    itk::simple::WriteImage(itk::simple::Image const &,std::string const &,bool)
    itk::simple::WriteImage(itk::simple::Image const &,std::vector< std::string,std::allocator< std::string > > const &,bool)

Как я могу решить эту проблему?

EDIT: Согласно ответу @Dženan, я получил следующее:

Traceback (most recent call last):
  File "rigid_transform.py", line 8, in <module>
    sitk.WriteImage(elastixImageFilter.GetResultImage(), 'result.jpg')
  File "/usr/local/lib/python2.7/dist-packages/SimpleElastix-1.0.1rc1.dev331+gd756f-py2.7-linux-x86_64.egg/SimpleITK/SimpleITK.py", line 8015, in WriteImage
    return _SimpleITK.WriteImage(*args)
RuntimeError: Exception thrown in SimpleITK WriteImage: /home/me/Desktop/SimpleElastix/build/ITK/Modules/IO/JPEG/src/itkJPEGImageIO.cxx:454:
itk::ERROR: JPEGImageIO(0x1a5daa0): JPEG supports unsigned char/int only

Спасибо.


python c++ simpleitk itk
person Simplicity    schedule 22.08.2017    source источник
comment
Этот вопрос теперь выглядит как дубликат stackoverflow.com/questions/45570341/   -  person blowekamp    schedule 28.08.2017

Ответы (2)


arrow_upward
1
arrow_downward

Вам не хватает имени файла. Попробуй это:

sitk.WriteImage(elastixImageFilter.GetResultImage(), 'result.jpg')

Редактировать: вы можете попробовать другие расширения файлов: result.png, result.tif, result.nrrd, result.mha и т. д.

person Dženan    schedule 22.08.2017
comment
Спасибо за добрый ответ. Пожалуйста, смотрите мое редактирование в вопросе. - person Simplicity; 22.08.2017
comment
Вы правы, использование других форматов файлов решает проблему. Например, PNG не работал, но другие работали. В настоящее время я пытаюсь НИИ. Большое спасибо. - person Simplicity; 23.08.2017

arrow_upward
2
arrow_downward

Как указано в прототипах С++, вам не хватает второго аргумента, имени файла.

person blowekamp    schedule 22.08.2017
comment
Спасибо за добрый ответ. Пожалуйста, смотрите мое редактирование в вопросе. - person Simplicity; 22.08.2017