diff --git a/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.h b/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.h
index 041319118c5..c18d499cd17 100644
--- a/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.h
+++ b/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.h
@@ -104,8 +104,7 @@ class ITK_TEMPLATE_EXPORT RGBGibbsPriorFilter : public MRFImageFilter<TInputImag
   using IndexType = typename TInputImage::IndexType;
 
   /** Set the image required for training type classifiers. */
-  void
-  SetTrainingImage(TrainingImageType image);
+  itkSetMacro(TrainingImage, TrainingImageType);
 
   /** Set the labelled image. */
   void
diff --git a/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.hxx b/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.hxx
index fb39850f880..f21ede1617c 100644
--- a/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.hxx
+++ b/Modules/Segmentation/MarkovRandomFieldsClassifiers/include/itkRGBGibbsPriorFilter.hxx
@@ -179,14 +179,6 @@ RGBGibbsPriorFilter<TInputImage, TClassifiedImage>::SetClassifier(typename Class
   m_ClassifierPtr->SetNumberOfClasses(m_NumberOfClasses);
 }
 
-/* Set the training image. */
-template <typename TInputImage, typename TClassifiedImage>
-void
-RGBGibbsPriorFilter<TInputImage, TClassifiedImage>::SetTrainingImage(TrainingImageType image)
-{
-  m_TrainingImage = image;
-}
-
 /* Check if 2 number are identical. */
 template <typename TInputImage, typename TClassifiedImage>
 int