Dr Yifan Liu's repositories
structure_knowledge_distillation
The official code for the paper 'Structured Knowledge Distillation for Semantic Segmentation'. (CVPR 2019 ORAL) and extension to other tasks.
CoupleGenerator
Generate your lover with your photo
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.
Auto_painter_demo
The code of building a web demo for Auto_painter
reid-strong-baseline
Bag of Tricks and A Strong Baseline for Deep Person Re-identification
antialiased-cnns
Antialiasing cnns to improve stability and accuracy. In ICML 2019.
DeepLabV3Plus-Pytorch
DeepLabv3, DeepLabv3+ and pretrained weights on VOC & Cityscapes
detectron2
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
guided-filter-pytorch
PyTorch implementation of Guided Image Filtering
diffuser
Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"
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.
mmdetection-distiller-CWD
This is a knowledge distillation toolbox based on mmdetection.
NRD_decoder
Dynamic Neural Representational Decoders for High-Resolution Semantic Segmentation
PointRend-simple-pytorch
a pytorch-based simple PointRend structure
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
Pytorch-UNet
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
vision
Datasets, Transforms and Models specific to Computer Vision