Skip to content

Latest commit

 

History

History
91 lines (61 loc) · 4.62 KB

README.md

File metadata and controls

91 lines (61 loc) · 4.62 KB

OneButtonDeepLearning

This repo provides the Graphic User Interface for Deep Learning models to realize "One-button" to use these models. In detail, it provides plugins in ImagePy to run the DL models.

让深度学习算法触手可及、一键调用,不必每次在命令行进行复杂配置。

Usage

Download the model folder, and place it in the imagepy/plugins folder. Then the menu for this model will appear in the imagepy menu bar.

只需将要使用的模型文件夹复制到imagepy/plugins文件夹下,再次启动ImagePy后即可在菜单栏看到该算法。

Optional

If the environment is not configured for the model, just enter its menus folder, and run:

pip install -r requirements.txt

Current available models

OCR / 光学字符识别

PaddleOCR aims to create multilingual, awesome, leading, and practical OCR tools that help users train better models and apply them into practice.

ocr-demo

YOLOv5 / 目标检测

YOLOv5 is a family of compound-scaled object detection models trained on the COCO dataset.

yolov5-demo

Image Matting / 抠图

Portrait Matting / 人物抠图

RVM is specifically designed for robust human video matting. Unlike existing neural models that process frames as independent images, RVM uses a recurrent neural network to process videos with temporal memory. RVM can perform matting in real-time on any videos without additional inputs. It achieves 4K 76FPS and HD 104FPS on an Nvidia GTX 1080 Ti GPU.

rvm-demo

Universal Matting / 通用抠图

FBA Matting is an interactive matting. It needs a trimap as an input, which can be easily realized with the ImagePy software.

fba-demo

Salient Object Detection / 显著目标检测

U2-Net does not need a trimap to detect the salient object in the image. sod

Face Detection / 人脸识别

InsightFace is an open source 2D&3D deep face analysis toolbox, and efficiently implements a rich variety of state of the art algorithms of face recognition, face detection and face alignment, which optimized for both training and deployment.

face-demo

Face Swap / 换脸

SimSwap is An Efficient Framework For High Fidelity Face Swapping.

faceswap-demo

Portrait Generation / 人物素描

U2-Net can also be used for human portrait drawing.

portrait

Cellpose / 胞状物体分割

Cellpose is a generalist algorithm for cell and nucleus segmentation.

cellpose-demo

Image Restoration / 图像重建

Real-ESRGAN / 超分辨

Real-ESRGAN aims at developing Practical Algorithms for General Image Restoration.

esrgan-demo esrgan-demo1

Bringing Old Photos Back to Life / 超分辨 + 去划痕

Old Photo Restoration is for Old Photo Restoration via Deep Latent Space Translation. It can be used to restore fuzzy images and even remove scratches.

oldimage-demo1 oldimage-demo2

Mask Inpainting / 魔法消除

LaMa is Resolution-robust Large Mask Inpainting with Fourier Convolutions.

lama

DeepLab / 语义分割

DeepLab is a state-of-art deep learning model for semantic image segmentation, where the goal is to assign semantic labels (e.g., person, dog, cat and so on) to every pixel in the input image. deeplab-demo