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
Related projects: ⓘ
- Utils for computer vision research.☆71Updated 5 years ago
- To increase the consistency of video segmentation, we add a temporal loss based on the optical flow.☆16Updated 5 years ago
- implement SfSNet using pytorch☆20Updated 2 years ago
- This is the source code of "Temporally Coherent Completion of Dynamic Video", ACM Transactions on Graphics (TOG), 2016, from https://file…☆16Updated 5 years ago
- ☆62Updated 3 years ago
- ☆42Updated 3 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Updated 4 years ago
- BoundarySqueeze: Image Segmentation as Boundary Squeezing☆54Updated 2 years ago
- Fast, high accuracy video segmentation framework (CVPR 2019 oral)☆53Updated 5 years ago
- Codes for "Attribute-Driven Spontaneous Motion in Unpaired Image Translation" (ICCV2019)☆24Updated 4 years ago
- ☆28Updated this week
- Develop and research with PyTorch more easily.☆25Updated 5 years ago
- [CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives☆84Updated 4 years ago
- Adopting Path Aggregation Network for Monocular Depth Estimation - PyTorch☆26Updated 6 years ago
- ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark☆21Updated 5 years ago
- SparseMask: Differentiable Connectivity Learning for Dense Image Prediction.☆51Updated 5 years ago
- Proposal, Tracking and Segmentation (PTS): A Cascaded Network for Video Object Segmentation☆118Updated 4 years ago
- This is the code for GMIS, which is published on ECCV 2018 as "Pixel Derivation and Graph Merge for Instance Segmentation"☆39Updated 5 years ago
- An asynchronous pytorch Dataloader for general neural network pipeline accelaration.☆53Updated 8 months ago
- ☆23Updated 3 years ago
- Code for CVPR 19 paper: MHP-VOS: Multiple Hypotheses Propagation for Video Object Segmentation☆62Updated 4 years ago
- deformable_conv2d layer implemented in pytorch☆64Updated 5 years ago
- Code for Paper "Bidirectional Aggregation Network with Occlusion Handling for Panoptic Segmentation"☆16Updated 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
- Semi-supervised Adaptive Distillation is a model compression method for object detection.☆58Updated 4 years ago
- ☆53Updated this week
- Code for our CVPR 2019 work.☆142Updated 5 years ago
- (ICCV2019) Learning Compositional Neural Infomation Fusion for Human Parsing☆37Updated last month
- The Pytorch implementation of "Location-aware Upsampling for Semantic Segmentation" (LaU)☆52Updated 4 years ago