An open-source toolkit which is full of handy functions, including the most used models and utilities for deep-learning practitioners!

PyTorch implementations of image classification networks

心律失常检测/ECG信号分类

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO

Automatization of the SEN-et SNAP plugin for multi-temporal analysis

Official PyTorch Implementation of Revisiting Self-Similarity: Structural Embedding for Image Retrieval, CVPR 2023

Code for paper "Channel Pruning Guided by Spatial and Channel Attention for DNNs in Intelligent Edge Computing"

pytorch implementations of some DL architectures. Includes googlenet, resnet, inceptionV3, densenet, mobilenetV1, mobilenetV2, senet, efficientnetV1, transformer etc.

Auto Assess Building Damage Based on Siamese Neural Network

Classification models trained on ImageNet. Keras.

Implementation of DSSM for recommendation by PyTorch (Including SENet and random negative sampling)

Implementation of our paper entitiled S&CNet: A lightweight network for fast and accurate depth completion published in JVCI.

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

可以训练yolov5(v6.0)、yolox、小型网络，添加注意力机制

Convert Caffe models to ONNX.

Leibniz is a python package which provide facilities to express learnable partial differential equations with PyTorch

動漫插畫放大/降噪

This repository is my journey towards learning ML which contains all my coding experiments ... :)

PyTorch implementation of SENet

PyTorch implementation of CNNs for CIFAR benchmark