Dr Yifan Liu's repositories
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
vision
Datasets, Transforms and Models specific to Computer Vision
diffuser
Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"
mmdetection-distiller-CWD
This is a knowledge distillation toolbox based on mmdetection.
NRD_decoder
Dynamic Neural Representational Decoders for High-Resolution Semantic Segmentation
Image_Harmonization_Datasets
Benchmark datasets and code used in our paper "DoveNet: Deep Image Harmonization via Domain Verification", CVPR2020. Useful for Image harmonization, image composition/compositing, etc.
Pytorch-UNet
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
DeepLabV3Plus-Pytorch
DeepLabv3, DeepLabv3+ and pretrained weights on VOC & Cityscapes
ETC-Real-time-Per-frame-Semantic-video-segmentation
Enforcing temporal consistency in real-time per-frame semantic video segmentation
Auto_painter
Recently, realistic image generation using deep neural networks has become a hot topic in machine learning and computer vision. Such an image can be generated at pixel level by learning from a large collection of images. Learning to generate colorful cartoon images from black-and-white sketches is not only an interesting research problem, but also a useful application in digital entertainment. In this paper, we investigate the sketch-to-image synthesis problem by using conditional generative adversarial networks (cGAN). We propose a model called auto-painter which can automatically generate compatible colors given a sketch. Wasserstein distance is used in training cGAN to overcome model collapse and enable the model converged much better. The new model is not only capable of painting hand-draw sketch with compatible colors, but also allowing users to indicate preferred colors. Experimental results on different sketch datasets show that the auto-painter performs better than other existing image-to-image methods.
EdgeNets
This repository contains the source code of our work on designing efficient CNNs for computer vision
FCHarDNet
Fully Convolutional HarDNet for Segmentation in Pytorch
ESPNetv2
A light-weight, power efficient, and general purpose convolutional neural network
AdelaiDet
AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
CaNet
The code for paper "CANet: Class-Agnostic Segmentation Networks with Iterative Refinement and Attentive Few-Shot Learning"
Auto_painter_demo
The code of building a web demo for Auto_painter
PointRend-simple-pytorch
a pytorch-based simple PointRend structure
structure_knowledge_distillation
The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.
guided-filter-pytorch
PyTorch implementation of Guided Image Filtering
detectron2
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
CoupleGenerator
Generate your lover with your photo
reid-strong-baseline
Bag of Tricks and A Strong Baseline for Deep Person Re-identification
CARAFE
Unofficial implementation of CARAFE: Content-Aware ReAssembly of FEatures. Pure pytorch imp
antialiased-cnns
Antialiasing cnns to improve stability and accuracy. In ICML 2019.