B
    >ex                 @   sP  d dl Z d dlZd dlZd dlmZ dd ZedkrLd dlZd dlm	Z	 e Z
dZdZdZd	Zd
dddgZejeZe ZeZeee
 dZede xedD ]Ze dZe edZe de e ee j Z!e de! eeee!Z"e Z#ee#e dZ$ee#e dZ%e#Zeee$e%e"e&e" e 'dd@ e(dkrP qW dS )    N)kerasc       	      C   s`   g }| | t|}|}||jd dddd}tj| |dd}|d }|| }|S )Nr         float32)axis)appendnparrayreshapeshapeastypeargmaxpredict)	modelLABimgtesttestaX_testpredp1L1 r   /home/pi/PW24/ML/trackh5.pymodh5   s    

 r   __main__)sleep   r   zmodel_NUM4C20.h51234   ztrackh5r.py as a main, tmodel=d   zNUM2_001.png)r   r   r   imggr      q))cv2numpyr   os
tensorflowr   r   __name__timer   tinincatisxisymod1r   models
load_modelr   t00t02roundtmodelprintrangejimreadr   resizeimg2imshowcvtColorCOLOR_BGR2GRAYr$   r   t01tdttypewaitKeyordr   r   r   r   <module>   s@   
   

   