文件名称:pytorch-receptive-field:一行计算火炬中CNN接收场的大小
文件大小:5KB
文件格式:ZIP
更新时间:2024-06-12 15:31:18
deep-learning cnn pytorch receptive-field Python
火炬接受野 在pytorch中计算CNN接收字段的大小 用法 git clone https://github.com/Fangyh09/pytorch-receptive-field.git from torch_receptive_field import receptive_field receptive_field ( model , input_size = ( channels , H , W )) 或者 from torch_receptive_field import receptive_field dict = receptive_field ( model , input_size = ( channels , H , W )) receptive_field_for_unit ( receptive_field_dict , "2" , ( 2 , 2 ))
【文件预览】:
pytorch-receptive-field-master
----torch_receptive_field()
--------__init__.py(99B)
--------receptive_field.py(8KB)
----main.py(841B)
----.travis.yml(124B)
----README.md(3KB)
----.gitignore(6B)