sonack / How_to_write_cuda_extensions_in_pytorch
How to write cuda kernels or c functions in pytorch, especially for former caffe users.
☆54Updated 6 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
- ☆63Updated 4 years ago
- Utils for computer vision research.☆72Updated 6 years ago
- Codes for "Attribute-Driven Spontaneous Motion in Unpaired Image Translation" (ICCV2019)☆24Updated 5 years ago
- ☆42Updated 3 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Updated 4 years ago
- ☆91Updated 4 years ago
- Explicit Shape Encoding for Real-time Instance Segmentation☆105Updated 2 years ago
- Semi-supervised Adaptive Distillation is a model compression method for object detection.☆59Updated 5 years ago
- SparseMask: Differentiable Connectivity Learning for Dense Image Prediction.☆51Updated 5 years ago
- implement SfSNet using pytorch☆21Updated 2 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
- Proposal, Tracking and Segmentation (PTS): A Cascaded Network for Video Object Segmentation☆117Updated 5 years ago
- Code for Paper "Bidirectional Aggregation Network with Occlusion Handling for Panoptic Segmentation"☆16Updated 4 years ago
- Video Generation from Single Semantic Label Map☆67Updated 5 years ago
- ☆47Updated 3 years ago
- Fast, high accuracy video segmentation framework (CVPR 2019 oral)☆54Updated 5 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Updated 4 years ago
- Learnable Tree Filter for Structure-preserving Feature Transform☆141Updated 2 years ago
- This is the source code of "Temporally Coherent Completion of Dynamic Video", ACM Transactions on Graphics (TOG), 2016, from https://file…☆15Updated 5 years ago
- An asynchronous pytorch Dataloader for general neural network pipeline accelaration.☆53Updated last year
- 3D Semantic Scene Completion☆9Updated 5 years ago
- EmbedMask: Embedding Coupling for One-stage Instance Segmentation☆160Updated 3 years ago
- ☆57Updated 4 years ago
- C++ extension implementation of RoIAlign & RolPool (both GPU and CPU) for PyTorch☆71Updated 6 years ago
- Adopting Path Aggregation Network for Monocular Depth Estimation - PyTorch☆26Updated 6 years ago
- Not All Pixels Are Equal: Difficulty-Aware Semantic Segmentation via Deep Layer Cascade☆108Updated 6 years ago
- ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark☆21Updated 5 years ago
- The Pytorch implementation of "Location-aware Upsampling for Semantic Segmentation" (LaU)☆53Updated 4 years ago
- Sparse Switchable Normalization with sparse activation function SparestMax☆64Updated 5 years ago
- Develop and research with PyTorch more easily.☆25Updated 6 years ago