PCMS-AI/MSP
0
1a
2 q��d+ � @ s* d Z ddlZd3dd�Zdd� Zdd � ZdS ) zG4 * The Inference of RAM and Tag2Text Models5 * Written by Xinyu Huang6� N�Nonec C s� t �� �& |j| d ddd�\}}W d � n1 s40 Y |dksV|dksV|dkrh|d d |d fS g }|�|�dd �� t �� �& |j| |ddd�\}}W d � n1 s�0 Y |d |d |d fS d S )7N�2 T)Z tag_input�8max_lengthZreturn_tag_predict� �noner r �,z | )�torch�no_grad�generate�append�replace)�image�modelZ input_tag�captionZ tag_predictZinput_tag_list� r �+d:\recognize-anything-main\ram\inference.py�inference_tag2text s" 9�(10�(r c C sF t �� � |�| �\}}W d � n1 s,0 Y |d |d fS �Nr )r r Z generate_tag)r
r �tags� tags_chineser r r �
inference_ram! s 11,r c C s: t �� � |�| �}W d � n1 s(0 Y |d S r )r r Zgenerate_tag_openset)r
r r r r r �inference_ram_openset) s 12(r )r )�__doc__r r r r r r r r �<module> s 13