YINYIPENG-EN / vit_classification_pytorch
采用vit实现图像分类
☆23Updated last year
Alternatives and similar repositories for vit_classification_pytorch:
Users that are interested in vit_classification_pytorch are comparing it to the libraries listed below
- 多模态数据融合:为了完成多模态数据融合,首先利用VGG16网络和cifar10数据集完成多输入网络的分类,在VGG16的基础之上,将前三层特征提取网络作为不同输入的特征提取网络,在中间层进行特征拼接,后面的卷积层用于提取融合特征,最后加上全连接层。该网络稍作修改就能同时提取…☆90Updated 4 years ago
- 基于Swin-transformer训练图像分类并部署web端☆90Updated 2 years ago
- 可视化神经网络☆38Updated 2 years ago
- Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https…☆89Updated 3 years ago
- 使用pytorch完成的一个多模态分类任务,文本和图像部分分别使用了bert和resnet提取特征(在config里可以组合多种模型),在我的小规模数据集上取得了良好的性能(验证集acc96%)☆77Updated 2 years ago
- 使用pytorch复现vit模型(图像分类)☆28Updated 2 years ago
- deep learning for image processing including classification and object-detection etc.☆25Updated 3 years ago
- This repo contains implementation of deep learning-based steel surface defect segmentation models. Extensive experiments on several deep …☆15Updated last year
- 利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆30Updated 4 years ago
- Awesome Fine-grained Visual Classification☆227Updated last year
- 这是一个clip-pytorch的模型,可以训练自己的数据集。☆223Updated 2 years ago
- 可视化特征图教程☆74Updated 3 years ago
- 基于PyTorch框架实现的图像分类网络☆80Updated 4 years ago
- ResNet50 猫狗数据集训练☆30Updated 4 years ago
- The official PyTorch implementation of our paper "MCA: Multidimensional collaborative attention in deep convolutional neural networks for…☆62Updated last year
- image classifier implement in pytoch.☆112Updated 2 years ago
- HiFuse: Hierarchical Multi-Scale Feature Fusion Network for Medical Image Classification☆230Updated 3 months ago
- [IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".☆85Updated last month
- [CIBM] Accurate Leukocyte Detection Based on Deformable-DETR and Multi-Level Feature Fusion for Aiding Diagnosis of Blood Diseases☆59Updated last year
- pytorch resnet image-classification☆19Updated 2 years ago
- DA-TransUNet: Combining Dual Attention of Position and Channel with Transformer U-net for Medical Image Segmentation☆152Updated 6 months ago
- 图像分类网络Pytorch实现☆12Updated 3 years ago
- SwinPA-Net: Swin Transformer-Based Multiscale Feature Pyramid Aggregation Network for Medical Image Segmentation☆24Updated last year
- ☆41Updated last year
- ☆34Updated 7 months ago
- Pytorch ViT for Image classification on the CIFAR10 dataset☆38Updated 3 years ago
- 这里包含了Mamba的代码以及b站对应的讲解视频☆84Updated 11 months ago
- Source code of the paper Multi-Granularity Part Sampling Attention for Fine-Grained Visual Classification☆27Updated 3 months ago
- 深度学习/计算机视觉/多模态/机器学习/人工智能零基础理论/实战教程汇总分享☆142Updated 2 years ago
- The official pytorch implementation of "SCSA: Exploring the Synergistic Effects Between Spatial and Channel Attention".☆75Updated last month