o
    Ďi                     @  sV   d dl mZ d dlZd dlmZ d dlmZ G dd dZddd	d
Z	ddddZ
dS )    )annotationsN)path)Pathc                   @  s$   e Zd ZdZdZdZdZdZdZdS )Yolov5TestConstantszGhttps://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5n.ptz#tests/data/models/yolov5/yolov5n.ptzHhttps://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s6.ptz$tests/data/models/yolov5/yolov5s6.ptzHhttps://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5m6.ptz$tests/data/models/yolov5/yolov5m6.ptN)	__name__
__module____qualname__YOLOV5N_MODEL_URLYOLOV5N_MODEL_PATHYOLOV5S6_MODEL_URLYOLOV5S6_MODEL_PATHYOLOV5M6_MODEL_URLYOLOV5M6_MODEL_PATH r   r   M/home/jeff/fluffinator/venv/lib/python3.10/site-packages/sahi/utils/yolov5.pyr      s    r   destination_path
str | Nonec                 C  D   | d u rt j} t| jjddd t| s tj	t j
|  d S d S NT)parentsexist_ok)r   r
   r   parentmkdirr   existsurllibrequesturlretriever	   r   r   r   r   download_yolov5n_model      
r   c                 C  r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   download_yolov5s6_model    r   r    )N)r   r   )
__future__r   urllib.requestr   osr   pathlibr   r   r   r    r   r   r   r   <module>   s    