瀏覽代碼

Add TFDWConv() `depth_multiplier` (#7858)

Enabled grouped non c1 == c2 convolutions in TF YOLOv5 models.
modifyDataloader
Glenn Jocher GitHub 2 年之前
父節點
當前提交
1b8e70f037
沒有發現已知的金鑰在資料庫的簽署中 GPG 金鑰 ID: 4AEE18F83AFDEB23
共有 1 個檔案被更改,包括 2 行新增1 行删除
  1. +2
    -1
      models/tf.py

+ 2
- 1
models/tf.py 查看文件

@@ -91,9 +91,10 @@ class TFDWConv(keras.layers.Layer):
def __init__(self, c1, c2, k=1, s=1, p=None, act=True, w=None):
# ch_in, ch_out, weights, kernel, stride, padding, groups
super().__init__()
assert c1 == c2, f'TFDWConv() input={c1} must equal output={c2} channels'
assert c2 % c1 == 0, f'TFDWConv() output={c2} must be a multiple of input={c1} channels'
conv = keras.layers.DepthwiseConv2D(
kernel_size=k,
depth_multiplier=c2 // c1,
strides=s,
padding='SAME' if s == 1 else 'VALID',
use_bias=not hasattr(w, 'bn'),

Loading…
取消
儲存