[PyTorch] Deformable Convolution示例
import torch
import torch.nn as nn
import torchvision.ops as ops
class Net(nn.Module):
def __init__(self, in_c, out_c, k=3):
super().__init__()
p = (k - 1) // 2
self.conv_offset = nn.Conv2d(in_c, 2 * k * k, k, padding=p)
.
原创
2021-11-24 16:47:35 ·
2793 阅读 ·
0 评论