mirror of
https://github.com/gaomingqi/Track-Anything.git
synced 2025-12-16 16:37:58 +01:00
14 lines
1.7 KiB
Plaintext
14 lines
1.7 KiB
Plaintext
|
a
|
|||
|
|
_{3d
|
|||
|
|
<00> @ sT d dl Z d dlZd dlZd dlmZ dd<04> Zd
ee d<06>dd<08>Zd d
|
|||
|
|
<EFBFBD> Z d d <0C> Z
|
|||
|
|
dS )<0E> N)<01>Optionalc C s | j d }| jdd<03>} |d ur0|jdd<03><01>d<02>nd }|jdd<03>}|d urT|jdd<03>nd }|d ur<>| <00>dd<02>} | <00>d<02>| }d| || }||<02>d<02> jddd<05>}| | | }n4| <00>d<02><01>d<01><01>d<02>}d| <00>dd<02>| }| | }|d u<01>r|| t<06>|<04> }n|t<06>|<04> }|S )N<> <00> )<01> start_dimT)<01>keepdim)<08>shape<70>flatten<65> unsqueeze<7A> transpose<73>pow<6F>sum<75>math<74>sqrt) <09>mk<6D>ms<6D>qk<71>qeZCKZa_sqZtwo_abZb_sq<73>
|
|||
|
|
similarity<EFBFBD> r <00>1/home/gaomingqi/SAM/xmem-sam/model/memory_util.py<70>get_similarity s$
|
|||
|
|
|
|||
|
|
|
|||
|
|
r F)<01>top_kc
|
|||
|
|
C s<> |d urjt j| |dd<02>\}}|<04><02> }|t j|ddd<04> }|rT| <00><04> <00>d||<06> | }q<>t <00>| <00><01>d||<06>}n>t j| ddd<04>d }t <00>| | <00>}t j|ddd<04>} || }d }|r<>||jdd<07>fS |S )Nr )<02>k<>dimT)r r r r )r ) <09>torch<63>topk<70>exp_r <00>zero_<6F>scatter_<72>
|
|||
|
|
zeros_like<EFBFBD>max<61>exp)
|
|||
|
|
r r <00>inplaceZ return_usage<67>values<65>indicesZx_exp<78>affinityZmaxesZ x_exp_sumr r r <00>
|
|||
|
|
do_softmax) s r& c C s t | |||<03>}t|<04>}|S <00>N)r r& )r r r r r r% r r r <00> get_affinityC s r( c C sF |j \}}}}}|<01>|||| | <00>}t<02>|| <00>}|<08>||||<06>}|S r' )r <00>viewr <00>bmm) r% <00>mv<6D>BZCV<43>T<>H<>W<>mo<6D>memr r r <00>readoutI s
|
|||
|
|
r2 )NFF) r
<00>numpy<70>npr <00>typingr r <00>intr& r( r2 r r r r <00><module> s "
|