Skip to content

Latest commit

 

History

History
47 lines (34 loc) · 1.3 KB

README.md

File metadata and controls

47 lines (34 loc) · 1.3 KB

收录到PytorchNetHub

说明

  • 非官方(官方Paddle版暂未开源)

  • 2020.12.15 修复bug,分组数s支持自定义

  • 2020.11.16 新增小尺寸输入

  • 2020.11.13 参数较ResNet增多,而非减少

    文中式(3)(4)有误,故无法得出HS-ResNet参数及计算量更少的结论

  • 2020.11.12 基于Pytorch官方ResNet,更省内存,支持更多网络

  • 2020.11.11 简单版本,内存占用稍高

环境

python版本 pytorch版本 系统
3.6 1.6.0 Ubuntu

支持网络

  • hs_resnet50
  • hs_resnet101
  • hs_resnet152
  • hs_resnext50_32x4d
  • hs_resnext101_32x8d
  • hs_wide_resnet50_2
  • hs_wide_resnet101_2

使用

import hs_resnet50
# w h>=224
net=hs_resnet50()
# w h<=112
net=hs_resnet50(small_input=True)
pred=net(imgs)

算法框架