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:
- 基于PyTorch框架实现的图像分类网络☆85Updated 5 years ago
- 基于Swin-transformer训练图像分类并部署web端☆94Updated 3 years ago
- (1) Purpose: A weakly supervised surface defect detection model using image-level labels for simultaneous classification and segmentation…☆23Updated 2 years ago
- 这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。☆45Updated 3 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆257Updated 4 months ago
- Unofficial implementations of some classical CNN backbones with pytorch☆167Updated 4 years ago
- ☆103Updated last year
- ☆48Updated last year
- pytorch实现MOAT,可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术☆34Updated 3 years ago
- ☆307Updated 4 years ago
- 论文复现,多机多卡☆49Updated 3 years ago
- pytorch resnet image-classification☆20Updated 3 years ago
- ☆96Updated 4 years ago
- 目标检测☆112Updated 5 years ago
- 基于无监督学习,无需训练,先使用SIFT算法提取图像特征,再使用KMeans聚类算法进行图像分类,速度非常快,精度还在提高。☆134Updated 6 years ago
- CNN经典网络结构复现:LeNet5、AlexNet、VGG、ResNet、InceptionNet等。☆102Updated 4 years ago
- 基于Pytorch的图像分类框架☆100Updated 4 months ago
- 利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆31Updated 5 years ago
- The official code of "Rethinking Local Perception in Lightweight Vision Transformer"☆91Updated 2 years ago
- fpn_resnet、resnet-se、resnet-cbam☆33Updated 5 years ago
- The official PyTorch implementation of our paper "MCA: Multidimensional collaborative attention in deep convolutional neural networks for…☆72Updated 2 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆265Updated 4 years ago
- YOLOv5-ODConvNeXt is an improved version of YOLOv5 for ship detection on drone-captured images.☆61Updated 2 years ago
- 玻璃绝缘子缺陷检测☆34Updated 5 years ago
- 这是一个blip-pytorch简化的代码,适用于了解Attention与Transformer的结构。☆55Updated 2 years ago
- ☆102Updated last year
- A python implementation of Canny Detector using numpy / scipy / torch / paddle package.☆29Updated last year
- Official repository of paper titled "CAS-ViT: Convolutional Additive Self-attention Vision Transformers for Efficient Mobile Applications…☆85Updated 3 weeks ago
- EfficientDet训练水下目标检测数据集☆49Updated 4 years ago
- ☆157Updated 4 years ago