pytorch-receptive-field:一行计算火炬中CNN接收场的大小

时间:2024-06-12 15:31:18
【文件属性】:

文件名称: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)

网友评论