sheep1222

detr

End-to-End Object Detection with Transformers

attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".

Swin-Unet

[ECCVW 2022] The codes for the work "Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation"

TransUNet

This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.

Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

A-Novel-Two-stage-Separable-Deep-Learning-Framework-for-Practical-Blind-Watermarking

The codes for the paper "A Novel Two-stage Separable Deep Learning Framework for Practical Blind Watermarking"

segmenter

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation

SegFormer

Official PyTorch implementation of SegFormer

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Pseudo-Differentiable-JPEG

Official repository for the paper "Towards Robust Data Hiding Against (JPEG) Compression: A Pseudo-Differentiable Deep Learning Approach".

pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet

pywt

PyWavelets - Wavelet Transforms in Python

MBRS

This is the source code of paper MBRS : Enhancing Robustness of DNN-based Watermarking by Mini-Batch of Real and Simulated JPEG Compression, which is received by ACM MM' 21.

HiNet

Official PyTorch implementation of "HiNet: Deep Image Hiding by Invertible Network" (ICCV 2021)

PerceptualSimilarity

LPIPS metric. pip install lpips

Convolutional-Neural-Network-Based-Image-Watermarking-using-Discrete-Wavelet-Transform

Deep-Model-Watermarking

ARWGAN

This repository is the official PyTorch implementation of ARWGAN: attention-guided robust image watermarking model based on GAN.

HiDDeN

Pytorch implementation of paper "HiDDeN: Hiding Data With Deep Networks" by Jiren Zhu, Russell Kaplan, Justin Johnson, and Li Fei-Fei

End-To-End-Image-Steganography

Replicated paper: END-TO-END TRAINED CNN ENCODER-DECODER NETWORKS FOR IMAGE STEGANOGRAPHY

SSRIW

Tensorflow implementation of 'Robust Image Watermarking based on Cross-Attention and Invariant Domain Learning'

DWSF

code repository for Practical Deep Dispersed Watermarking with Synchronization

ReDMark

automatic-watermark-detection

Project for Digital Image Processing

ultralytics

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

ImageProcessing-and-MachineLearning

图像处理、机器学习的常用算法汇总

python-bchlib

BCH library C Python module

StegaStamp

Invisible Hyperlinks in Physical Photographs