A collection of Python scripts designed to interface with the Lumerical Design Suite.

PyTorch implementation for Histogram Loss

This repository contains code for paper "Learning Deep Embeddings with Histogram Loss" (NIPS2016)

PyTorch implementation for differentiable histogram

Easy generative modeling in PyTorch.

Tutorials and code collections of my publications associated with photonic neural networks.

A PyTorch Library for Photonic Integrated Circuit Simulation and Photonic AI Computing

Memory consumption and FLOP count estimates for convnets

Optical diffraction neural network on PyTorch

D2NN TRANNING SCRIPT

Kolmogorov Arnold Networks

[CVPR 2022--Oral, Best paper Finalist] Burst Image Restoration and Enhancement. SOTA for Burst Super-resolution, Low-light Burst Image Enhancement, Burst Image De-noising

SOTA for Burst Super-resolution, Low-light Burst Image Enhancement, Burst Image De-noising

Official implementation of Deep Burst Super-Resolution

A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Repository for Neural Network project 2023/2024, Ai & Robotics, cascaded Conditional Gan for map to satellite view generation.

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

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

High Quality Monocular Depth Estimation via Transfer Learning

[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

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

Optical Neural Networks on PyTorch. diffractive propagation, nonlinear-photonic-activation

Pytorch Unofficial implement of paper "All optical machine learning using diffractive deep neural networks" .

Diffraction Deep Neural Networks(D2NN)