Wenjie Zhang's repositories
algotom
Data processing algorithms for tomography: https://algotom.readthedocs.io
AutoKernel
AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。
Awesome-Image-Colorization
:books: A collection of Deep Learning based Image Colorization and Video Colorization papers.
BigColor
[ECCV 2022] Official PyTorch implementation of "BigColor"
csi-nn2
An optimized neural network operator library for chips base on Xuantie CPU.
CV-CUDA
CV-CUDA™ is an open-source, graphics processing unit (GPU)-accelerated library for cloud-scale image processing and computer vision.
DALLE2-pytorch
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
ddrm
[NeurIPS 2022] Denoising Diffusion Restoration Models -- Official Code Repository
DiffBIR
Official codes of DiffBIR: Towards Blind Image Restoration with Generative Diffusion Prior
facechain
FaceChain is a deep-learning toolchain for generating your Digital-Twin.
fastjson
FASTJSON 2.0.x has been released, faster and more secure, recommend you upgrade.
Hands-On-GPU-Accelerated-Computer-Vision-with-OpenCV-and-CUDA
Hands-On GPU Accelerated Computer Vision with OpenCV and CUDA, published by Packt
highway
Performance-portable, length-agnostic SIMD with runtime dispatch
image-segmentation-keras
Implementation of Segnet, FCN, UNet , PSPNet and other models in Keras.
imgaug
Image augmentation for machine learning experiments.
MNN
MNN is a blazing fast, lightweight deep learning framework, battle-tested by business-critical use cases in Alibaba
once-for-all
[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment
only_train_once
[ICLR 2023] OTOv2: Automatic, Generic, User-Friendly; [NeurIPS 2021] Only Train Once: A One-Shot Neural Network Training And Pruning Framework
Open3D
Open3D: A Modern Library for 3D Data Processing
RWKV-CUDA
The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )
RWKV-LM
RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.
sarepy
Numerical techniques for ring artifact removal in tomography.
sedna
AI tookit over KubeEdge
Simd
C++ image processing and machine learning library with using of SIMD: SSE, AVX, AVX-512, AMX for x86/x64, VMX(Altivec) and VSX(Power7) for PowerPC, NEON for ARM.
stable-diffusion-webui
Stable Diffusion web UI
Tengine
Tengine is a lite, high performance, modular inference engine for embedded device
TIGRE
TIGRE: Tomographic Iterative GPU-based Reconstruction Toolbox
xray
List of datasets and papers in X-ray security images (Computer vision/Machine Learning)
xsimd
C++ wrappers for SIMD intrinsics and parallelized, optimized mathematical functions (SSE, AVX, AVX512, NEON, SVE))