dcn_v2_onnx.py make.sh setup.py testcpu.py testcuda.py README BSD-3-Clause license DCNv2 latest Since DCN is used in many models and performance well but in industry this op support is not very well. Including pytorch, onnx, tensorrt etc. This repo is make DCNv2 available at all ...
pokameng/DCNv2_latestPublic forked fromlucasjinreal/DCNv2_latest NotificationsYou must be signed in to change notification settings Fork0 Star0 Code Pull requests Actions Projects Security Insights Additional navigation options Files master src
DCNv2 latest Add support for pytorch1.11~pytorch1.13 (may be not backward-compatible). Test0 on Ubuntu20.04~Ubuntu22.04.3 Test1 on Python3.8(Conda)~Python3.9.13(Conda) Test2 on CUDA_11.4~CUDA_11.7 It was confirmed that Pytorch1.11~Pytorch1.13 worked, but not compatible with previous pytorch ...
dcn_v2_conv = _DCNv2.apply class DCNv2(nn.Module): def __init__( self, in_channels, out_channels, kernel_size, stride, padding, dilation=1, deformable_groups=1, ): super(DCNv2, self).__init__() self.in_channels = in_channels self.out_channels = out_channe...
dcn_v2.py dcn_v2_onnx.py make.sh setup.py testcpu.py testcuda.py DCNv2 latest Add support for pytorch1.11 (may be not backward-compatible). Test on ubuntu20.04, python3.8(conda), cuda_11.4 It was confirmed that pytorch1.11 worked, but not compatible with previous pytorch version. If ...
class _DCNv2(Function): @staticmethod def symbolic(g, input, offset_mask, weight, bias, stride, padding, dilation, deformable_groups): return g.op("Plugin", input, offset_mask, weight, bias, name_s="DCNv2", info_s=json.dumps({ "dilation": dilation, "padding": padding, "st...
dcn_v2_conv = _DCNv2.apply class DCNv2(nn.Module): def __init__(self, in_channels, out_channels, kernel_size, stride=1, padding=1, dilation=1, deformable_groups=1): super(DCNv2, self).__init__() self.in_channels = in_channels self.out_channels = out_channels self....
This branch is 13 commits behind lucasjinreal/DCNv2_latest:master.Folders and files Latest commit Your Name update for pytorch 1.7+ support d6971f7· Feb 17, 2021 History52 Commits src update for pytorch 1.7+ support Feb 17, 2021 .gitignore rm dist Dec 11, 2018 LICENSE some license ^^^...
DCNv2 supports decent pytorch such as torch 1.5+ (now 1.7+) - DCNv2_latest/README.md at master · pokameng/DCNv2_latest
This branch is13 commits behindlucasjinreal/DCNv2_latest:master. Folders and files Name Last commit message Last commit date Latest commit Cannot retrieve latest commit at this time. History 52 Commits src .gitignore LICENSE README.md __init__.py ...