Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
迪士尼在逃法务96
efficientnet-pytorch
提交
7e8b0d31
E
efficientnet-pytorch
项目概览
迪士尼在逃法务96
/
efficientnet-pytorch
与 Fork 源项目一致
从无法访问的项目Fork
通知
4
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
E
efficientnet-pytorch
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
7e8b0d31
编写于
4月 15, 2021
作者:
L
Luke Melas-Kyriazi
提交者:
GitHub
4月 15, 2021
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #250 from rvandeghen/patch-1
Add new checkpoint
上级
45834ee9
75ca1bf5
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
14 addition
and
12 deletion
+14
-12
efficientnet_pytorch/model.py
efficientnet_pytorch/model.py
+14
-12
未找到文件。
efficientnet_pytorch/model.py
浏览文件 @
7e8b0d31
...
...
@@ -238,18 +238,18 @@ class EfficientNet(nn.Module):
Returns:
Dictionary of last intermediate features
with reduction levels i in [1, 2, 3, 4, 5].
Example:
>>> import torch
>>> from efficientnet.model import EfficientNet
>>> inputs = torch.rand(1, 3, 224, 224
)
>>> model = EfficientNet.from_pretrained('efficientnet-b0'
)
>>> endpoints = model.extract_endpoints(inputs
)
>>> print(endpoints['reduction_1'].shape) # torch.Size([1, 16, 112, 112
])
>>> print(endpoints['reduction_2'].shape) # torch.Size([1, 24, 56, 56
])
>>> print(endpoints['reduction_3'].shape) # torch.Size([1, 40, 28, 28
])
>>> print(endpoints['reduction_4'].shape) # torch.Size([1, 112, 14, 14
])
>>> print(endpoints['reduction_5
'].shape) # torch.Size([1, 1280, 7, 7])
Example:
>>> import torch
>>> from efficientnet.model import EfficientNet
>>> inputs = torch.rand(1, 3, 224, 224)
>>> model = EfficientNet.from_pretrained('efficientnet-b0'
)
>>> endpoints = model.extract_endpoints(inputs
)
>>> print(endpoints['reduction_1'].shape) # torch.Size([1, 16, 112, 112]
)
>>> print(endpoints['reduction_2'].shape) # torch.Size([1, 24, 56, 56
])
>>> print(endpoints['reduction_3'].shape) # torch.Size([1, 40, 28, 28
])
>>> print(endpoints['reduction_4'].shape) # torch.Size([1, 112, 14, 14
])
>>> print(endpoints['reduction_5'].shape) # torch.Size([1, 320, 7, 7
])
>>> print(endpoints['reduction_6
'].shape) # torch.Size([1, 1280, 7, 7])
"""
endpoints
=
dict
()
...
...
@@ -265,6 +265,8 @@ class EfficientNet(nn.Module):
x
=
block
(
x
,
drop_connect_rate
=
drop_connect_rate
)
if
prev_x
.
size
(
2
)
>
x
.
size
(
2
):
endpoints
[
'reduction_{}'
.
format
(
len
(
endpoints
)
+
1
)]
=
prev_x
elif
idx
==
len
(
self
.
_blocks
)
-
1
:
endpoints
[
'reduction_{}'
.
format
(
len
(
endpoints
)
+
1
)]
=
x
prev_x
=
x
# Head
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录