Datasets
Models
Downloads: 1
45 lines (45 with data), 6.1 kB
öíú_─ Ń @ sn d d l Z d d l m Z m Z d d l Z d d l m Z Gd d ä d e â Z Gd d ä d e â Z
d S)Ú N)┌io┌color)┌trangec @ sO e Z d Z d Z d d d d d ä Z d d ä Z d d ä Z d d
ä Z d S) ┌ClusterÚ r c C sA | j | | | | | â g | _ | j | _ t j d 7_ d S)Nr )┌update┌pixels┌
cluster_index┌nor )┌self┌h┌w┌l┌a┌bę r ˙I/raid/data/wangqiuli/Documents/YH_Interpretable/Version1/features/slic.py┌__init__
s zCluster.__init__c C s1 | | _ | | _ | | _ | | _ | | _ d S)N)r r
r r r )r r r
r r r r r r r s
zCluster.updatec C s( d j | j | j | j | j | j â S)Nz{},{}:{} {} {} )┌formatr r
r r r )r r r r ┌__str__ s zCluster.__str__c C s
| j â S)N)r )r r r r ┌__repr__ s zCluster.__repr__N)┌__name__┌
__module__┌ __qualname__r r r r r r r r r r s
r c @ sú e Z d Z e d d ä â Z e d d ä â Z d d ä Z d d ä Z d d
ä Z d d ä Z d
d ä Z
d d ä Z d d ä Z d d ä Z
d d d ä Z d S)┌
SLICProcessorc C s/ t j | â } t j | â } t | j â | S)z=
Return:
3D array, row col [LAB]
)r ┌imreadr Zrgb2lab┌print┌shape)┌path┌rgb┌lab_arrr r r ┌
open_image s
zSLICProcessor.open_imagec C s# t j | â } t j | | â d S)z}
Convert the array to RBG, then save the image
:param path:
:param lab_arr:
:return:
N)r Zlab2rgbr ┌imsave)r r Zrgb_arrr r r ┌save_lab_image* s z SLICProcessor.save_lab_imagec C sC t | | | j | | d | j | | d | j | | d â S)Nr r Ú )r ┌data)r r r
r r r ┌ make_cluster5 s zSLICProcessor.make_clusterc C s░ | | _ | | _ | | _ | j j d | _ | j j d | _ | j | j | _ t t j | j | j â â | _
g | _ i | _ t
j | j | j f t
j â | _ d S)Nr r )┌K┌Mr% r ┌ image_height┌ image_width┌N┌int┌math┌sqrt┌S┌clusters┌label┌np┌full┌inf┌dis)r ┌imgr' r( r r r r ; s " zSLICProcessor.__init__c C sŹ | j d } | j d } xl | | j k rł x< | | j k rj | j j | j | | â â | | j 7} q/ W| j d } | | j 7} q Wd S)Nr$ )r/ r) r* r0 ┌appendr& )r r r
r r r ┌
init_clustersI s
zSLICProcessor.init_clustersc C sđ | d | j k r | j d } | d | j k r@ | j d } | j | d | d d | j | | d | j | d | d d | j | | d | j | d | d d | j | | d } | S)Nr r$ r )r* r) r% )r r r
┌gradientr r r ┌ get_gradientS s
îzSLICProcessor.get_gradientc
C sŔ xß | j D]Í } | j | j | j â } xÁ t d d â D]Ą } xŤ t d d â D]Ő } | j | } | j | } | j | | â } | | k rN | j | | | j | | d | j | | d | j | | d â | } qN Wq8 Wq
Wd S)Nr r$ r Ú r; )r0 r: r r
┌ranger r% )r ┌clusterZcluster_gradientZdh┌dw┌_h┌_wZ new_gradientr r r ┌
move_clusters^ s
FzSLICProcessor.move_clustersc
C sQ xJ| j D]?} x6t | j d | j | j d | j â D] } | d k s< | | j k r` q< xÔt | j d | j | j d | j â D]Á} | d k sî | | j k r░ qî | j | | \ } } } t j t j
| | j d â t j
| | j d â t j
| | j
d â â } t j t j
| | j d â t j
| | j d â â } t j t j
| | j d â t j
| | j d â â } | | j | | k rî | | f | j k rń| | j | | f <| j j | | f â nL | j | | f j j | | f â | | j | | f <| j j | | f â | | j | | <qî Wq< Wq
Wd S)Nr$ r )r0 r< r r/ r) r
r* r% r- r. ┌powr r r r( r5 r1 r r7 ┌remove)
r r= r r
┌L┌A┌BZDcZDs┌Dr r r ┌
assignmentj s. 2 2 - 9#zSLICProcessor.assignmentc C s║ x│ | j D]Ę } d } } } xĹ | j D]ć } | | d 7} | | d 7} | d 7} | | } | | } | j | | | j | | d | j | | d | j | | d â q( Wq
Wd S)Nr r r$ )r0 r r r% )r r= Zsum_hZsum_w┌number┌pr? r@ r r r ┌update_clusterâ s
z SLICProcessor.update_clusterc C sĽ t j | j â } x| | j D]q } xh | j D]] } | j | | d | d d <| j | | d | d d <| j | | d | d d <q, Wq W| S)Nr r r$ )r2 ┌copyr% r0 r r r r )r ┌nameZ image_arrr= rJ r r r ┌save_current_imageÄ s !z SLICProcessor.save_current_image┌tempc C sč | j â | j â g } xz t d â D]l } | j â | j â | d t | â d ů d j d | d | j d | j â } | j | j
| â â q' W| d S) NÚ Ú zM{m}_K{k}_loop{loop}.png┌loop┌m┌kr r; ) r8 rA r< rH rK ┌lenr r( r' r7 rN )r ┌filenameZtemp_img┌irM r r r ┌iterate_10timesč s
<z SLICProcessor.iterate_10timesN)r r r ┌ staticmethodr! r# r& r r8 r: rA rH rK rN rX r r r r r s
r ) r- ┌skimager r ┌numpyr2 ┌tqdmr ┌objectr r r r r r ┌<module> s