a
    j0$e                     @   s|   d dl T d dlZd dlZd dlZd dlmZ d dlmZ d dlZ	d dl
mZ d dlmZ d dlZejejdZdd	 ZdS )
    )*N)keras)
device_lib)Image)settingsz,MachineLearningModels/AppleDetectionModel.ptc              
      s   t jjddtd}| jj}||}|  t| t	dD ]}d| d tj
 rvt fddt	 D }nd	}z&tjj| j| jd
}||_|  W n, tjy   tjj| j| j|dd Y n0 td| d |   q<d S )Nzultralytics/yolov5custom)pathzruns/detectzruns/detect/z/crops/applesc                    s&   g | ]}t jt j |r|qS  )osr   isfilejoin).0entry	directoryr	   K/var/www/html/django/DPS/api/functionsNeuralNetworks/appleDetectionModel.py
<listcomp>"       z'appleDetectionModel.<locals>.<listcomp>r   )tree_id
mission_idT)r   r   yieldshas_scab/)torchhubload
appleModel	image_urir   cropprintr
   listdirexistslen	TreeStateobjectsgetr   r   r   saveDoesNotExistcreateshutilrmtreedelete)ZphotomodelimgresultsfolderZamountOfPearsstater	   r   r   appleDetectionModel   s$    r1   )Zgardens.modelsr   r
   Z
tensorflowtfr   Ztensorflow.python.clientr   numpynpPILr   django.confr   r)   r   r   STATIC_ROOTr   r1   r	   r	   r	   r   <module>   s   