年龄估计模型 SSR-Net 模型解析
1. 论文模型图:
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。
模型架构图:
细节:
图中的各个模块的细节请参考开源代码中的具体实现。
https://github.com/oukohou/SSR_Net_Pytorch/blob/master/SSR_models/SSR_Net_model.py
注:pytorch 中的 view 在 onxx 的可视化 Netron 中 Reshape;Linear 为 Gemm;切片 为 Gather。
模型架构图:
细节:
图中的各个模块的细节请参考开源代码中的具体实现。