

Title: ResMambaBlock 模块技术细节图Layout: 极简横向流,从左 Input 到右 Output。1. External Flow起点:Input X (B, C, H, W)。两路:主路径 + 灰色虚线弧形残差(Identity)。主路径依次两个蓝色虚线框容器:Block 1:Norm(燕麦色矩形)→ ReLU(浅橙圆)→ Mamba Layer 1(莫兰迪蓝,展开)。Block 2:Norm(燕麦色)→ ReLU(浅橙)→ Mamba Layer 2(莫兰迪蓝,简化)。最右红色 [+] 汇合主路径与残差 → Output。2. Mamba Layer 1 展开维度变换1:浅紫圆角 Reshape & Transpose,🔥 (B, C, H, W) → (B, N, C),N=H×W。LayerNorm(燕麦色,fp32),🔥红圈标数值稳定。内部双支路:下支:深蓝 Mamba SSM Core(d_state=16, d_conv=4, expand=2)。上支:skip_scale(可学习)。红色 [+] 融合。后处理:LayerNorm(燕麦)→ Linear Proj (C→C,深蓝)。维度变换2:浅紫 Transpose & Reshape,🔥 (B, N, C) → (B, C, H, W)。3. Style背景 #F7F5F2。颜色:Proj深蓝 #5B7C99;维度操作浅紫 #C5B9D6;Norm #F4E8C1;Mamba #8FA3B5;ReLU #FFB347;加法 #E74C3C。细深灰线,残差灰色虚线箭头弧。字体 Arial/Helvetica,维度小号斜体。
描述你想生成的图片信息,如:画面里有什么风格/元素/色彩
Shift+回车 = 换行
AI可能会犯错误或不完全遵循,请谨慎使用。
同款 0
评论 0