B
    jbej                 @   sT  d dl Zd dlZdd Zdd ZedkrPd dlmZ ed e	d	d
dgZ
e	dddgZe	d d
dgZe	dddgZdZed Zeejd eejd e \ZZed eee
eeee\ZZZedeeee eedZede ed  eee
eeee\ZZedeee eedZede ed  dS )    Nc             C   s  |j d d }t|}t|tj}t|||}	| dkrTt|||}
|	|
B }n
t|	}tj|||d}t|}t|tj	tj
\}}t|}d}ddddddg}t|}|dkrt|tjd}t|}t|}t||ddd t|\}}}}t|||f|| || fdd}|t|d  }|t|d  }t|||fddd}t|d	 }d|  krd
k rn nd}nBd
|  krdkrn nd}n d|  k rdk rn nd}|| }||||||g}ntd t|||f}|||fS )Nr         )mask)key)r      r   )r   r   r            2   P      zlencontour=0)shapenpcopycv2cvtColorCOLOR_BGR2HSVinRangebitwise_andfindContoursRETR_EXTERNALCHAIN_APPROX_NONElenmaxcontourAreadrawContoursboundingRect	rectangleintcircleprinthstack)MSKlow1upp1low2upp2imgYimg1hsvmask1mask3mask2img2threshcontours	hierarchyimg3TDRboxlencontrcxcaxxywhZcyZX8ZAxh2 r>   /home/pi/PW24/targfunc.pytargx   sL    





  
 

"     r@   c             C   s  |j d d }t|}t|tj}t|||}	| dkrTt|||}
|	|
B }n
t|	}tj|||d}t|}t|tj	tj
\}}t|}ddddg}t|}|dkrt|tjd}t|}d}g }x|D ]}t|}|dk rq|d7 }t||ddd t|\}}}}t|||f|| || fd	d}|| }||||||g}|| t|| qW td
|d| ntd t|||f}||fS )Nr   r   r   )r   )r   <   r   )r   r   r   )r   r   r   z	lencontr=zmax_c_area=zlencontour=0)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   appendr!   r"   )r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r5   r6   r7   r8   ncboxesccar9   r:   r;   r<   Ar=   r>   r>   r?   targs*   sH    





 
 
 

  "  
rH   __main__)sleepztargx as a main      r      r   
   r   i  i  z
targx:TDR=)i     Ztargx_h2ztargs:Ztargs_h2)numpyr   r   r@   rH   __name__timerJ   r!   arrayr$   r%   r&   r'   r#   VideoCapturecapsetCAP_PROP_FRAME_WIDTHCAP_PROP_FRAME_HEIGHTreadretr(   r4   r5   r=   r   resizeimshowwaitKeyrD   r>   r>   r>   r?   <module>   s4   &%


