jiantenggei / torch-classification
用来自己训练和复现分类模型
☆18Updated 2 years ago
Alternatives and similar repositories for torch-classification:
Users that are interested in torch-classification are comparing it to the libraries listed below
- 基于PyTorch框架实现的图像分类网络☆80Updated 4 years ago
- 利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆30Updated 4 years ago
- 基于Swin-transformer训练图像分类并部署web端☆90Updated 2 years ago
- [CVPR 2022] TVConv: Efficient Translation Variant Convolution for Layout-aware Visual Processing☆46Updated 2 years ago
- 资源受限环境下、大规模肺炎早筛方法。采用DSHNet生成少类样本数据,解决数据不平衡的问题,然后利用RSFNet进行分类,最后结合剪枝策略实现轻量化!MedGAN-ResLite-V2 is released! ❤☆27Updated 10 months ago
- 深度学习 注意力机制模块 时间 空间注意力通道☆29Updated last year
- 玻璃绝缘子缺陷检测☆32Updated 4 years ago
- Context-aware Cross-level Fusion Network for Camouflaged Object Detection☆51Updated 2 years ago
- pytorch resnet image-classification☆19Updated 2 years ago
- ☆34Updated last year
- 论文复现,多机多卡☆49Updated 2 years ago
- Cosine Annealing Warm Booting LR☆27Updated 3 years ago
- DPNet: Dual-Path Network for Real-time Object Detection with Lightweight Attention☆18Updated 2 years ago
- Official repository of paper titled "CAS-ViT: Convolutional Additive Self-attention Vision Transformers for Efficient Mobile Applications…☆56Updated 4 months ago
- (1) Purpose: A weakly supervised surface defect detection model using image-level labels for simultaneous classification and segmentation…☆23Updated last year
- object detection dataset conversion scripts☆23Updated 2 years ago
- ☆18Updated last year
- 可视化特征图教程☆74Updated 3 years ago
- pytorch实现MOAT,可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术☆31Updated 2 years ago
- ☆40Updated last year
- ResNet50 猫狗数据集训练☆30Updated 4 years ago
- 深度学习常用的脚本,比如公共数据集(Tusimple,VOC等)的转换(xml2json,json2xml,yolo2voc),读取视频,分离训练验证集等☆20Updated 3 years ago
- The official pytorch implementation of "SCSA: Exploring the Synergistic Effects Between Spatial and Channel Attention".☆73Updated last month
- Pytorch实践简单, 方便, 快速训练 图像分类模型☆37Updated 3 years ago
- 适合小白入门的图像分类项目,从熟悉到熟练图像分类的流程,搭建自己的分类网络结构以及在pytorch中运用经典的分类网络。☆67Updated 5 years ago
- 基于Pytorch实现优秀的自然图像分割框架!(包括FCN、U-Net和Deeplab)☆65Updated 3 years ago
- 这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。☆42Updated 2 years ago
- 数据增强仓库Albumentations的使用☆59Updated 4 years ago
- 各种算法评价指标的实现(mAP/Flops/params/fps/error-rate/accuracy)☆31Updated 9 months ago
- 细粒度图像分类之十二猫分类,对比ResNet和ViT两者模型性能。☆44Updated 3 years ago