sonack / How_to_write_cuda_extensions_in_pytorch
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
- 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
- Learnable Tree Filter for Structure-preserving Feature Transform☆141Updated 2 years ago
- Utils for computer vision research.☆72Updated 6 years ago
- ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark☆22Updated 5 years ago
- A flexible pytorch DataParallel module☆22Updated 6 years ago
- Fast, high accuracy video segmentation framework (CVPR 2019 oral)☆54Updated 5 years ago
- implement SfSNet using pytorch☆21Updated 2 years ago
- SparseMask: Differentiable Connectivity Learning for Dense Image Prediction.☆51Updated 6 years ago
- Sparse Switchable Normalization with sparse activation function SparestMax☆64Updated 5 years ago
- ☆42Updated 3 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Updated 4 years ago
- deformable_conv2d layer implemented in pytorch☆62Updated 6 years ago
- ☆64Updated 4 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Updated 4 years ago
- Code accompanying the paper Optimizing the F-measure for Threshold-free Salient Object Detection.☆29Updated 5 years ago
- PyTorch Implementation of the paper Learning Affinity via Spatial Propagation Networks☆63Updated 7 years ago
- ☆91Updated 4 years ago
- Proposal, Tracking and Segmentation (PTS): A Cascaded Network for Video Object Segmentation☆117Updated 5 years ago
- Not All Pixels Are Equal: Difficulty-Aware Semantic Segmentation via Deep Layer Cascade☆108Updated 6 years ago
- Explicit Shape Encoding for Real-time Instance Segmentation☆105Updated 2 years ago
- Develop and research with PyTorch more easily.☆25Updated 6 years ago
- Data Priming Network for Automatic Check-Out - ACMMM 2019☆26Updated 5 years ago
- Codes for "Attribute-Driven Spontaneous Motion in Unpaired Image Translation" (ICCV2019)☆24Updated 5 years ago
- ☆47Updated 3 years ago
- A simple, fast, efficient and end-to-end 3D object detector without NMS.☆30Updated 3 years ago
- Dense reppoints: Representing visual objects with dense point sets https://arxiv.org/abs/1912.11473☆145Updated 4 years ago
- Code for "Jointly Optimize Data Augmentation and Network Training: Adversarial Data Augmentation in Human Pose Estimation" (CVPR 2018)☆84Updated 6 years ago
- BoundarySqueeze: Image Segmentation as Boundary Squeezing☆56Updated 3 years ago
- ☆23Updated 4 years ago
- Code for our paper "Looking for the Devil in the Details: Learning Trilinear Attention Sampling Network for Fine-grained Image Recognitio…☆53Updated 3 years ago