Juartaurus/poi_Engineering
0
1U
2 Gk
e� � @ sl d dl Zd dlZd dlZd dlZd dlZd dlmZ e� Z dd� Z3dd� Z dd� Z d d4� Z
ed krhe
� dS ) � N)�SignboardTrackerc C s t j�| �st �| � d S )N)�os�path�isdir�makedirs)�dir_path� r �//home/kienvs/sources/poi_engineering_api/app.py�5create_dir6 s r7 c C s8 | d k r t j�| ��d�d }t j�d|� �}t j�d|� d��}t|� t|� t j�|t j�| ��}t�| �}t|� tj8��}t|� tj ��}t|� tj ��}tj
d� } t�t j�|t j�| ��| |||f�}9t j�dd|� dt j�| ��d�d � ��} d} |�� \}
}|
d k�r,|10�|� n�q,�q|| | |fS )11N�.r zstatic/videos/z/frames�mp4vzstatic/videos/{video_name}/�tracked_�����T)r r �basename�split�joinr12 �cv2� VideoCapture�int�get�CAP_PROP_FRAME_WIDTH�CAP_PROP_FRAME_HEIGHT� CAP_PROP_FPS�VideoWriter_fourcc� VideoWriter�read�write)� input_video�13video_name� video_dir�tracking_result_dir�vd_dir� video_cap� frame_width� frame_height�fps�fourcc�output� output_track�output_track_format�success�framer r r �get_meta_from_video s( 1415&, 16 r, c C s t �| |||�d }|S )Nr )�signboardTracker�inference_signboard)r! r( r) r �resultsr r r �tracking* s r0 c C s� t jdd���Z} t �d� t �� �� t jdd��� t jdd�}|� t jdd�}W 5 Q R X t jd d�}|�( t �� � t jd d17d �}W 5 Q R X W 5 Q R X t jd d�}|�: t �� �& t �� }t �� }t �� }t �� } W 5 Q R X W 5 Q R X W 5 Q R X t jd
d��� t �� �� t jdd��n t �� }18t �� �R t jdd�� t jdd19d �} W 5 Q R X t jdd�� t jdd20d �} W 5 Q R X W 5 Q R X W 5 Q R X t jdd��p t d�D ]`}
t �� �L t jd
d�� t �� }W 5 Q R X t jdd�� t jdd21d �}W 5 Q R X W 5 Q R X �q�W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X |j22t |g|||| gd� |j t
|||| g|gd� W 5 Q R X | jdd� | jd23d24d25d� d S )Nz*.gradio-container {background-color: gray})�cssz�26 <div style="text-align:center;">27 <span style="font-size:3em; font-weight:bold;">Segment and Track Anything(SAM-Track)</span>28 </div>29 g�������?)�scalezVideo type input)�labelz Input video�TrackingT)�value� interactive�save_dirg�������?g �?g �?z30Next ImagezPrevious Imageg �?�31 z
All Signboard)�fn�inputs�outputs� )�concurrency_count)�debug� enable_queue�share)�gr�Blocks�Markdown�Row�Column�Video�Button�Textbox�Image�range�changer, �clickr0 �queue�launch)�demo�tab_video_inputr �tab_everything�
seg_signboardr7 r! r( r) r �input_first_frame�next�previous�i�cropimg�allr r r �
seg_track_app- sj � 32" 33& 34, 35N� �� 36� rY �__main__)�numpy�np�gradiorA r r �shutil�src.tracker.signboard_trackr r- r37 r, r0 rY �__name__r r r r �<module> s O