Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
曾经的那一瞬间
Models
提交
804d6abc
M
Models
项目概览
曾经的那一瞬间
/
Models
11 个月 前同步成功
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
Models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
804d6abc
编写于
9月 09, 2021
作者:
V
Vishnu Banna
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
nn_blocks update
上级
f35907fd
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
19 addition
and
13 deletion
+19
-13
official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
...al/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
+19
-13
未找到文件。
official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
浏览文件 @
804d6abc
...
...
@@ -48,7 +48,7 @@ class ConvBN(tf.keras.layers.Layer):
strides
=
(
1
,
1
),
padding
=
'same'
,
dilation_rate
=
(
1
,
1
),
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -97,7 +97,14 @@ class ConvBN(tf.keras.layers.Layer):
self
.
_strides
=
strides
self
.
_padding
=
padding
self
.
_dilation_rate
=
dilation_rate
self
.
_kernel_initializer
=
kernel_initializer
if
kernel_initializer
==
"VarianceScaling"
:
# to match pytorch initialization method
self
.
_kernel_initializer
=
tf
.
keras
.
initializers
.
VarianceScaling
(
scale
=
1
/
3
,
mode
=
'fan_in'
,
distribution
=
'uniform'
)
else
:
self
.
_kernel_initializer
=
kernel_initializer
self
.
_bias_initializer
=
bias_initializer
self
.
_kernel_regularizer
=
kernel_regularizer
...
...
@@ -194,7 +201,7 @@ class DarkResidual(tf.keras.layers.Layer):
filters
=
1
,
filter_scale
=
2
,
dilation_rate
=
1
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
kernel_regularizer
=
None
,
bias_regularizer
=
None
,
...
...
@@ -366,7 +373,7 @@ class CSPTiny(tf.keras.layers.Layer):
def
__init__
(
self
,
filters
=
1
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -532,7 +539,7 @@ class CSPRoute(tf.keras.layers.Layer):
filters
,
filter_scale
=
2
,
activation
=
'mish'
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -661,7 +668,7 @@ class CSPConnect(tf.keras.layers.Layer):
drop_first
=
False
,
activation
=
'mish'
,
kernel_size
=
(
1
,
1
),
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -783,7 +790,7 @@ class CSPStack(tf.keras.layers.Layer):
model_to_wrap
=
None
,
filter_scale
=
2
,
activation
=
'mish'
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -796,7 +803,6 @@ class CSPStack(tf.keras.layers.Layer):
"""CSPStack layer initializer.
Args:
filters: integer for output depth, or the number of features to learn.
model_to_wrap: callable Model or a list of callable objects that will
process the output of CSPRoute, and be input into CSPConnect.
list will be called sequentially.
...
...
@@ -884,7 +890,7 @@ class PathAggregationBlock(tf.keras.layers.Layer):
def
__init__
(
self
,
filters
=
1
,
drop_final
=
True
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -1120,7 +1126,7 @@ class SAM(tf.keras.layers.Layer):
strides
=
(
1
,
1
),
padding
=
'same'
,
dilation_rate
=
(
1
,
1
),
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -1192,7 +1198,7 @@ class CAM(tf.keras.layers.Layer):
def
__init__
(
self
,
reduction_ratio
=
1.0
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -1285,7 +1291,7 @@ class CBAM(tf.keras.layers.Layer):
strides
=
(
1
,
1
),
padding
=
'same'
,
dilation_rate
=
(
1
,
1
),
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
@@ -1363,7 +1369,7 @@ class DarkRouteProcess(tf.keras.layers.Layer):
insert_cbam
=
False
,
csp_stack
=
0
,
csp_scale
=
2
,
kernel_initializer
=
'
glorot_uniform
'
,
kernel_initializer
=
'
VarianceScaling
'
,
bias_initializer
=
'zeros'
,
bias_regularizer
=
None
,
kernel_regularizer
=
None
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录