Commit a8184dc3 authored by Zhe Chen's avatar Zhe Chen Committed by zhe chen
Browse files

Update README.md and release models (#44)

* Update README.md

* update configs

* clean code

* support InternImage-H/G
parent 6be127ee
...@@ -187,6 +187,7 @@ class DCNv3_pytorch(nn.Module): ...@@ -187,6 +187,7 @@ class DCNv3_pytorch(nn.Module):
N, H, W, _ = input.shape N, H, W, _ = input.shape
x = self.input_proj(input) x = self.input_proj(input)
x_proj = x
x1 = input.permute(0, 3, 1, 2) x1 = input.permute(0, 3, 1, 2)
x1 = self.dw_conv(x1) x1 = self.dw_conv(x1)
...@@ -313,6 +314,7 @@ class DCNv3(nn.Module): ...@@ -313,6 +314,7 @@ class DCNv3(nn.Module):
N, H, W, _ = input.shape N, H, W, _ = input.shape
x = self.input_proj(input) x = self.input_proj(input)
x_proj = x
dtype = x.dtype dtype = x.dtype
x1 = input.permute(0, 3, 1, 2) x1 = input.permute(0, 3, 1, 2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment