sonack / How_to_write_cuda_extensions_in_pytorchLinks
How to write cuda kernels or c functions in pytorch, especially for former caffe users.
☆54Updated 7 years ago
Alternatives and similar repositories for How_to_write_cuda_extensions_in_pytorch
Users that are interested in How_to_write_cuda_extensions_in_pytorch are comparing it to the libraries listed below
Sorting:
- Utils for computer vision research.☆72Updated 7 years ago
- This is the code for GMIS, which is published on ECCV 2018 as "Pixel Derivation and Graph Merge for Instance Segmentation"☆39Updated 6 years ago
- ☆64Updated 5 years ago
- SparseMask: Differentiable Connectivity Learning for Dense Image Prediction.☆51Updated 6 years ago
- ☆42Updated 4 years ago
- deformable_conv2d layer implemented in pytorch☆63Updated 6 years ago
- Sparse Switchable Normalization with sparse activation function SparestMax☆64Updated 6 years ago
- Code for Paper "Bidirectional Aggregation Network with Occlusion Handling for Panoptic Segmentation"☆16Updated 5 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Updated 5 years ago
- Explicit Shape Encoding for Real-time Instance Segmentation☆104Updated 2 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Updated 5 years ago
- ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark☆22Updated 6 years ago
- Proposal, Tracking and Segmentation (PTS): A Cascaded Network for Video Object Segmentation☆117Updated 5 years ago
- This is the pytorch implementation of "Adaptively Connected Neural Networks" for the currently popular EfficientNet and the efficient DNA…☆10Updated 5 years ago
- A ShuffleBatchNorm layer to shuffle BatchNorm statistics across multiple GPUs☆56Updated 3 years ago
- You are welcomed to join us!☆51Updated 5 years ago
- ☆34Updated 2 years ago
- Adopting Path Aggregation Network for Monocular Depth Estimation - PyTorch☆26Updated 7 years ago
- Learnable Tree Filter for Structure-preserving Feature Transform☆142Updated 3 years ago
- Code for Switchable Whitening (ICCV2019)☆137Updated 5 years ago
- Unofficial implementation for SOLO instance segmentation☆25Updated 5 years ago
- Pytorch 1.0 codes(including cuda codes) for Deformable Convolution Version 2☆18Updated 6 years ago
- Dense reppoints: Representing visual objects with dense point sets https://arxiv.org/abs/1912.11473☆145Updated 5 years ago
- C++ extension implementation of RoIAlign & RolPool (both GPU and CPU) for PyTorch☆71Updated 6 years ago
- Package containing helper functions for loading and evaluating DAVIS☆66Updated 7 years ago
- BoundarySqueeze: Image Segmentation as Boundary Squeezing☆56Updated 3 years ago
- Semi-supervised Adaptive Distillation is a model compression method for object detection.