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
Sorting:
- 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
- Utils for computer vision research.☆72Updated 6 years ago
- SparseMask: Differentiable Connectivity Learning for Dense Image Prediction.☆51Updated 6 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Updated 4 years ago
- ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark☆22Updated 5 years ago
- This is the source code of "Temporally Coherent Completion of Dynamic Video", ACM Transactions on Graphics (TOG), 2016, from https://file…☆15Updated 6 years ago
- ☆64Updated 4 years ago
- Semi-supervised Adaptive Distillation is a model compression method for object detection.☆59Updated 5 years ago
- ☆42Updated 3 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
- Proposal, Tracking and Segmentation (PTS): A Cascaded Network for Video Object Segmentation☆117Updated 5 years ago
- implement SfSNet using pytorch☆21Updated 2 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
- Learnable Tree Filter for Structure-preserving Feature Transform☆141Updated 2 years ago
- Explicit Shape Encoding for Real-time Instance Segmentation☆105Updated 2 years ago
- An unofficial implementation of FCOS in Pytorch: 37.3-resnet50-700px☆57Updated 5 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Updated 4 years ago
- Sparse Switchable Normalization with sparse activation function SparestMax☆64Updated 5 years ago
- Dense reppoints: Representing visual objects with dense point sets https://arxiv.org/abs/1912.11473☆145Updated 4 years ago
- C++ extension implementation of RoIAlign & RolPool (both GPU and CPU) for PyTorch☆71Updated 6 years ago
- Code for CVPR 19 paper: MHP-VOS: Multiple Hypotheses Propagation for Video Object Segmentation☆62Updated 5 years ago
- Code accompanying the paper Optimizing the F-measure for Threshold-free Salient Object Detection.☆29Updated 5 years ago
- ☆91Updated 4 years ago
- deformable_conv2d layer implemented in pytorch☆62Updated 6 years ago
- Code for Paper "Bidirectional Aggregation Network with Occlusion Handling for Panoptic Segmentation"☆16Updated 5 years ago
- ☆34Updated 2 years ago
- A flexible pytorch DataParallel module☆22Updated 6 years ago
- Develop and research with PyTorch more easily.☆25Updated 6 years ago
- Pytorch implementation of Single-Image Depth Perception in the Wild https://arxiv.org/pdf/1604.03901.pdf☆38Updated last year