Files
Track-Anything/model/__pycache__/memory_util.cpython-39.pyc

14 lines
1.7 KiB
Plaintext
Raw Normal View History

2023-04-12 08:24:08 +08:00
a
_{3d
<00>@sTddlZddlZddlZddlmZdd<04>Zd eed<06>dd<08>Zd d
<EFBFBD>Z d d <0C>Z
dS)<0E>N)<01>Optionalc Cs|jd}|jdd<03>}|dur0|jdd<03><01>d<02>nd}|jdd<03>}|durT|jdd<03>nd}|dur<>|<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>|}| |}|du<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_similaritys$
   

rF)<01>top_kc
Cs<>|durjtj||dd<02>\}}|<04><02>}|tj|ddd<04>}|rT|<00><04><00>d||<06>|}q<>t<00>|<00><01>d||<06>}n>tj|ddd<04>d}t<00>||<00>}tj|ddd<04>} || }d}|r<>||jdd<07>fS|S)Nr)<02>k<>dimT)rrrr)r) <09>torch<63>topk<70>exp_r <00>zero_<6F>scatter_<72>
zeros_like<EFBFBD>max<61>exp)
rr<00>inplaceZ return_usage<67>values<65>indicesZx_exp<78>affinityZmaxesZ x_exp_sumrrr<00>
do_softmax)s r&cCst||||<03>}t|<04>}|S<00>N)rr&)rrrrrr%rrr<00> get_affinityCsr(c CsF|j\}}}}}|<01>|||||<00>}t<02>||<00>}|<08>||||<06>}|Sr')r<00>viewr<00>bmm) r%<00>mv<6D>BZCV<43>T<>H<>W<>mo<6D>memrrr<00>readoutIs
 r2)NFF) r <00>numpy<70>npr<00>typingrr<00>intr&r(r2rrrr<00><module>s "