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框架实现的图像分类网络☆75Updated 4 years ago
- pytorch实现MOAT,可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术☆29Updated 2 years ago
- A python implementation of Canny Detector using numpy / scipy / torch / paddle package.☆27Updated 8 months ago
- This repository is a PyTorch version of the paper "EWT: Efficient Wavelet-Transformer for Single Image Denoising"☆24Updated 7 months ago
- 可视化特征图教程☆65Updated 2 years ago
- ☆34Updated last year
- 利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆27Updated 4 years ago
- 深度学习 注意力机制模块 时间 空间注意力通道☆28Updated last year
- 基于Swin-transformer训练图像分类并部署web端☆85Updated 2 years ago
- Multi-Scale Adaptive Network for Single Image Denoising (NeurIPS 2022)☆11Updated 2 years ago
- 资源受限环境下、大规模肺炎早筛方法。采用DSHNet生成少类样本数据,解决数据不平衡的问题,然后利用RSFNet进行分类,最后结合剪枝策略实现轻量化!MedGAN-ResLite-V2 is released! ❤☆25Updated 7 months ago
- pytorch resnet image-classification☆18Updated 2 years ago
- 这是各个主干网络分类模型的源码,可以用于训练自己的分类模型。☆68Updated last year
- [CVPR 2022] TVConv: Efficient Translation Variant Convolution for Layout-aware Visual Processing☆43Updated 2 years ago
- [ICASSP 2024] This is the official code for "LEFormer: A Hybrid CNN-Transformer Architecture for Accurate Lake Extraction from Remote Sen…☆14Updated 9 months ago
- 各种算法评价指标的实现(mAP/Flops/params/fps/error-rate/accuracy)☆31Updated 5 months ago
- Compound Multi-branch Feature Fusion for Real Image Restoration☆93Updated 5 months ago
- 论文复现,多机多卡☆47Updated 2 years ago
- 细粒度图像分类之十二猫分类,对比ResNet和ViT两者模型性能。☆35Updated 2 years ago
- 这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。☆42Updated 2 years ago
- 可视化神经网络☆40Updated 2 years ago
- ☆27Updated 5 years ago
- ☆42Updated 3 months ago
- PyTorch implementation of ResNeSt : Split-Attention Networks☆18Updated 4 years ago
- ☆49Updated 5 months ago
- ☆153Updated 3 years ago
- Pytorch implement ion of RepMLP☆31Updated 3 years ago
- ☆13Updated 6 months ago
- 里面会保存许多优秀的卷 积神经网络结构,这些结构可以帮助我们更好的设计网络。☆142Updated 3 years ago