#!/usr/bin/env?python3
#?-*-?coding:?utf-8?-*-
“““
Created?on?Sat?Mar?10?16:04:33?2018

@author:?sy
“““



import?numpy?as?np
import?pydensecrf.densecrf?as?dense_crf
#from?.?import?potentials
from?densecrf2?import?potentials

class?DenseCRF:
????def?__init__（self?num_classes?zero_unsure?unary_potential?pairwise_potentials?use_2d?=?‘rgb-1d‘）:
????????“““A?wrapper?for?the?PyDenseCRF?functions.?
????????
????????Currently?only?2D?images?are?supported.
????????
????????Arguments
????????---------
????????num_classes?:?int
????????unary_potential?:?UnaryPotential
????????pairwise_potentials?:?PairwisePotential?or?array?like
????????????A?collection?of?PairwisePotential?instances.
????????“““
????????if?isinstance（pairwise_potentials?potentials.PairwisePotential）:
????????????pairwise_potentials?=?[pairwise_potentials]

????????for?pairwise_potential?in?pairwise_potentials:
????????????self.check_potential（pairwise_potential?potentials.PairwisePotential）
????????self.check_potential（unary_potential?potentials.UnaryPotential）

????????self.zero_unsure?=?zero_unsure
????????self.pairwise_potentials?=?pairwise_potentials
????????self.unary_potential?=?unary_potential
????????
????????self.use_2d?=?use_2d
????????self.num_classes?=?num_classes
????????self.crf_model?=?None
????????self._image?=?None
????????self._colour_axis?=?None

????????#?Variables?needed?for?the?CRF?model?to?run
????????self._Q?=?None
????????self._tmp1?=?None
????????self._tmp2?=?None

????def?set_image（self?image?probabilities?colour_axis=None?class_axis=None?label_source=‘label‘）:
????????“““Set?the?image?for?the?CRF?model?to?perform?inference?on.

????????Arguments
????????---------
????????image?:?numpy.ndarray
????????????The?image?to?segment.
????????probabilities?:?numpy.ndarray
????????????Class?probabilities?for?each?pixel.
????????colour_axis?:?int
????????????Which?axis?of?the?image?array?the?colour?information?lies?in.
????????????Usually?the?last?axis?（-1）?but?can?also?be?the?first?axis?（0）.
????????????If?‘image‘?is?a?grayscale?image?this?should?be?set?to?None.
????????class_axis?:?int
????????????Which?axis?of?the?probabilities?array?the?class?information?lies.
????????????If?the?probabilites?are?on?the?form
????????????????‘‘[image_height?image_width?class]‘‘
????????????then?class?axis?should?be?set?to?either?‘3‘?or?‘-1‘.
????????“““
????????if?colour_axis?is?None:
????????????colour_axis?=?-1
????????????image?=?image.reshape（*image.shape?1）

????????self._image?=?image
????????self._colour_axis?=?self.fix_negative_index（colour_axis）
????????self._class_axis?=?class_axis
????????self._probabilities?=?probabilities
????????self._image_shape?=?\
????????????image.shape[:self._colour_axis]?+?image.shape[self._colour_axis+1:]
????????self.label_source?=?label_source

##????????‘rgb-1d‘?or?‘rgb-2d‘?or?‘non-rgb‘
????????if?self.use_2d?==?‘rgb-1d‘:
????????????self.process_rgb_label（）
????????????self._create_model（）
????????????self._set_potentials_1d（）

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-03-12?09:38??CRF-semantic-segmentation-master\
?????文件????????1491??2018-03-12?09:38??CRF-semantic-segmentation-master\README.md
?????目錄???????????0??2018-03-12?09:38??CRF-semantic-segmentation-master\densecrf2\
?????文件??????????50??2018-03-12?09:38??CRF-semantic-segmentation-master\densecrf2\__init__.py
?????文件???????10298??2018-03-12?09:38??CRF-semantic-segmentation-master\densecrf2\crf_model.py
?????文件????????9681??2018-03-12?09:38??CRF-semantic-segmentation-master\densecrf2\potentials.py
?????目錄???????????0??2018-03-12?09:38??CRF-semantic-segmentation-master\examples\
?????文件????????3742??2018-03-12?09:38??CRF-semantic-segmentation-master\examples\non_rgb_image_test_1.py
?????文件????????3800??2018-03-12?09:38??CRF-semantic-segmentation-master\examples\non_rgb_image_test_2.py
?????文件????????3350??2018-03-12?09:38??CRF-semantic-segmentation-master\examples\rgb_image_test.py