回家倒数第6天
受不了了,不想写了,这群b怎么这么能写
用latex写了个伪代码,有点好玩
\usepackage[ruled,linesnumbered]{algorithm2e}
\begin{algorithm}[ht]
\caption{Pipeline of Kernel Iteration in K-Net.}
\label{alg:alg1}
\BlankLine
\KwIn{Number of update iteration stages, $N$; Mask prediction at the $i^{th}$ stage, $M_i$; Kernels at the $i^{th}$ stage, $K_i$; Image feature maps, $F$; Linear transformation, $\phi_j, j=\{1,2\}$; Fully connected layer, $\psi_k, k=\{1,2,3,4\}$.}
\KwOut{Mask prediction, $M_N$; Updated kernels, $K_N$.}
\BlankLine
Initialize $K_0$ randomly\;
\While{$i<N$}{
\textbf{Group Feature Assembling}\\
\qquad $F^K \leftarrow M_iF, F \in \mathbb{R}^{B \times N \times C}$\;
\textbf{Adaptive Feature Update}\\
\qquad $F^G \leftarrow \phi_1(F^K) \cdot \phi_2(K_{i-1})$\;
\qquad $G^K \leftarrow \sigma(\psi_1(F^G))$\;
\qquad $G^F \leftarrow \sigma(\psi_2(F^G))$\;
\qquad $\tilde{K} \leftarrow G^F \cdot \psi_3(F^K) + G^K \cdot \psi_4(K_{i-1})$\;
\textbf{Kernel Iteration}\\
\qquad $K_i \leftarrow Multi{-}head\;attention(\tilde{K})$\;
\qquad $M_i \leftarrow K_i * F$\;
$i \leftarrow i+1$\;
}
\emph{For simplicity, some operations are not written.}
\end{algorithm}
最后是长这样