年龄估计模型 SSR-Net 模型解析

1. 论文模型图:

image

2. 模型代码:pytorch 实现

https://github.com/oukohou/SSR_Net_Pytorch/blob/master/SSR_models/SSR_Net_model.py

注:pytorch 中的 view 在 onxx 的可视化 Netron 中 Reshape;Linear 为 Gemm;切片 为 Gather。

模型架构图:

image

细节:

image

image

图中的各个模块的细节请参考开源代码中的具体实现。

posted @ 2023-02-08 11:49  cold_moon  阅读(102)  评论(0编辑  收藏  举报