Adamfan/objectdetection
0
1�
2 0e:h � �b � d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd d�Zd d�Z d
d�Z3dd4�Z dS )� N)�Path� c �� � t j d| � �d�� � � d� � }t |� � r(t |� d� � d � � ndS )Nz5gsutil du T��shellzutf-8� r )�6subprocess� check_output�decode�len�eval�split)�url�ss �G/content/drive/MyDrive/demo_yolov7_torch26/yolov7/utils/google_utils.py�gsutil_getsizer
sZ � �� � 2�S� 2� 2�$�?�?�?�F�F�w�O�O�A�$'��F�F� 1�4����� � �Q�� � � ��1� �WongKinYiu/yolov7c � � t t | � � � � � � dd� � � � � � � } | � � � �sd t
j d|� d�� � � � � }d� |d D � � }|d }nK# g d�}t j7 d d8� � � � � � � � � d }Y nxY w| j
}||v �r�| � d
|� d�}d} d|� d|� d|� �}t d|� d| � d�� � t j � || � � | � � � r | � � � j dk sJ �nm# t( $ r`} t d| � �� � |s9J d� � �d|� d|� �}t d|� d| � d�� � t+ j d|� d | � �� � Y d } ~ nd } ~ ww xY w| � � � r | � � � j dk r(| � d10� � � t d |� �� � t d� � d S # | � � � r | � � � j dk r(| � d11� � � t d |� �� � t d� � Y d S xY wd S d S ) N�'r z https://api.github.com/repos/z/releases/latestc � � g | ]12}|d �� S )�name� )�.0�xs r �13<listcomp>z$attempt_download.<locals>.<listcomp> s � �<�<�<�A�a��i�<�<�<r �assets�tag_name)z yolov7.ptzyolov7-tiny.ptz14yolov7x.ptz yolov7-d6.ptz yolov7-e6.ptz
yolov7-e6e.ptz yolov7-w6.ptzgit tagTr �����z2 missing, try downloading from https://github.com/z15/releases/Fzhttps://github.com/z/releases/download/�/z Downloading z to z...g ��.AzDownload error: zNo secondary mirrorz https://storage.googleapis.com/z/ckpt/zcurl -L z -o ��16missing_okzERROR: Download failure: )r �str�strip�replace�lower�exists�requests�get�jsonr r17 r r r �print�torch�hub�download_url_to_file�stat�st_size� Exception�os�system�unlink)18�file�repo�responser �tagr �msg� redundantr �es19 r �attempt_downloadr< s2 � � ��D� � ���!�!�)�)�#�r�2�2�8�8�:�:� ;� ;�D� �;�;�=�=� � V� �|�$Z�D�$Z�$Z�$Z�[�[�`�`�b�b�H�<�<��(�);�<�<�<�F��:�&�C�C�� V�7� 7� 7�F� �)�)�4�@�@�@�G�G�I�I�O�O�Q�Q�RT�U�C�C�C�����y�� �6�>�>��]�]�T�]�]�]�C� �I�
�Q�D�Q�Q�S�Q�Q�4�Q�Q���7�S�7�7�d�7�7�7�8�8�8�� �.�.�s�D�9�9�9��{�{�}�}�B������)<�s�)B�)B�)B�B�)B�� �
6�
6�
6��,��,�,�-�-�-� �7�7�"7�7�7�y�J��J�J�D�J�J���7�S�7�7�d�7�7�7�8�8�8�� �4�S�4�4�d�4�4�5�5�5�5�5�5�5�5�����
6���� �{�{�}�}� =�� � � � �(;�c�(A�(A��K�K�4�K�0�0�0��;�c�;�;�<�<�<��b� � � ���� �{�{�}�}� =�� � � � �(;�c�(A�(A��K�K�4�K�0�0�0��;�c�;�;�<�<�<��b� � � ��������= � � �>sA �+AB0 �0AC8�A4F �I �20G1�AG,�'I �,G1�1I � A*K �tmp.zipc �f � t j � � }t |� � }t d� � }t d| � d|� d�d�� � |� d�� � |� d�� � t j � � d k rd21nd }t
j d | � d
|� �� � t j � d� � rdt � � � d| � d|� �}n d|� d| � d�}t
j |� � }|� d�� � |dk r'|� d�� � t d� � |S |j22 dk r<t dd�� � t
j d|� �� � |� � � t dt j � � |z23 d�d�� � |S ) N�cookiez;Downloading https://drive.google.com/uc?export=download&id=z as z... r )�endTr! �Windows�NULz /dev/nullz?curl -c ./cookie -s -L "drive.google.com/uc?export=download&id=z" > z?curl -Lb ./cookie "drive.google.com/uc?export=download&confirm=z&id=z" -o zcurl -s -L -o z) "drive.google.com/uc?export=download&id=�"r zDownload error z.zipz
unzipping... z unzip -q zDone (z.1fzs)) �timer r+ r4 �platformr3 r2 �pathr' � get_token�suffix)�idr5 �tr? �outr �rs r �gdrive_downloadrM 8 s� � � � � � �A� ��:�:�D�
�(�^�^�F� �24Z��25Z�26Z�PT�27Z�28Z�29Z�`b�c�c�c�c��K�K�4�K� � � �30�M�M�T�M�"�"�"� �?�$�$� �1�1�%�%�{�C��I�]�PR�]�]�X[�]�]�^�^�^� �w�~�~�h� � � R� n�i�k�k� n� n�_a� n� n�hl� n� n��� Q�T� Q� Q�B� Q� Q� Q��31� �!� � �A�32�M�M�T�M�"�"�"� �A�v�v� � � �t� �$�$�$�
� � � � ��� �{�f� � �
�o�2�&�&�&�&�33� �$�d�$�$�%�%�%� � � �
�
�
� �34*�4�9�;�;��?�35*�36*�37*�38*�+�+�+� �Hr �./cookiec � � t | � � 5 }|D ].}d|v r(|� � � d c cd d d � � S �/ d d d � � n # 1 swxY w Y dS )N�downloadr r )�openr )r? �f�lines r rG rG [ s� � �
�f� � � (��� (� (�D��T�!�!��z�z�|�|�B�'�'�'�(� (� (� (� (� (� (� (�!� (�(� (� (� (� (� (� (� (� (� (� (���� (� (� (� (� �2s �$A�A�A�A)r )r )r r= )rN ) r2 rE r rD �pathlibr r( r, r r< rM rG r r r �<module>rU s� �� 39� � � � ���� � � � � � � � � � � � � � � ���� � � � �2� 2� 2� 2� "� "� "� "�J
�
�
�
�F� � � � � r