jiantenggei / torch-classificationLinks
用来自己训练和复现分类模型
☆19Updated 3 years ago
Alternatives and similar repositories for torch-classification
Users that are interested in torch-classification are comparing it to the libraries listed below
Sorting:
- 基于Swin-transformer训练图像分类并部署web端☆94Updated 3 years ago
- 基于PyTorch框架实现的图像分类网络☆85Updated 4 years ago
- ResNet50 猫狗数据集训练☆33Updated 4 years ago
- pytorch resnet image-classification☆19Updated 3 years ago
- 细粒度图像分类之十二猫分类,对比ResNet和ViT两者模型性能。☆45Updated 3 years ago
- A python implementation of Canny Detector using numpy / scipy / torch / paddle package.☆29Updated last year
- Unofficial implementations of some classical CNN backbones with pytorch☆167Updated 4 years ago
- Official repository of paper titled "CAS-ViT: Convolutional Additive Self-attention Vision Transformers for Efficient Mobile Applications…☆75Updated 10 months ago
- 这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。☆46Updated 3 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆258Updated 2 weeks ago
- 论文复现,多机多卡☆49Updated 3 years ago
- ☆34Updated 2 years ago
- Compound Multi-branch Feature Fusion for Real Image Restoration☆101Updated last year
- 基于Pytorch的图像分类框架☆95Updated last month
- pytorch实现MOAT,可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术☆34Updated 3 years ago
- 深度学习 注意力机制模块 时间 空间注意力通道☆33Updated 2 years ago
- TPSeNCE for image rain generation, deraining, and object detection.☆105Updated 10 months ago
- ☆102Updated 9 months ago
- Unofficial PyTorch implementation of MobileViT based on paper "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Trans…☆125Updated 3 years ago
- 玻璃绝缘子缺陷检测☆33Updated 4 years ago
- 目标检测☆111Updated 5 years ago
- 使用CNN进行图像分类,并且附录了SIFT+BOG+SVM多分类器的机器学习分类模型☆40Updated 6 years ago
- [ICCV2023] This is an official implementation for "Scale-Aware Modulation Meet Transformer".☆211Updated 2 years ago
- 一款即插即用的知识蒸馏工具包☆11Updated 3 years ago
- 基于YOLOv5和PSPNet的实时目标检测和语义分割系统☆40Updated 3 years ago
- ACMMM2021 paper "I2V-GAN: Unpaired Infrared-to-Visible Video Translation"☆125Updated 3 years ago
- 这是一个blip-pytorch简化的代码,适用于了解Attention与Transformer的结构。☆53Updated last year
- Vision Transformers with Hierarchical Attention☆102Updated last month
- HSViT: Horizontally Scalable Vision Transformer☆13Updated 11 months ago
- 基于无监督学习,无需训练,先使用SIFT算法提取图像特征,再使用KMeans聚类算法进行图像分类,速度非常快,精度还在提高。☆133Updated 6 years ago