YINYIPENG-EN / vit_classification_pytorch
采用vit实现图像分类
☆19Updated 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的基础之上,将前三层特征提取网络作为不同输入的特征提取网络,在中间层进行特征拼接,后面的卷积层用于提取融合特征,最后加上全连接层。该网络稍作修改就能同时提取…☆81Updated 4 years ago
- 可视化神经网络☆40Updated 2 years ago
- 基于Swin-transformer训练图像分类并部署web端☆85Updated 2 years ago
- 使用pytorch完成的一个多模态分类任务,文本和图像部分分别使用了bert和resnet提取特征(在config里可以组合多种模型),在我的小规模数据集上取得了良好的性能(验证集acc96%)☆70Updated last year
- This repo is about multi-label medical image classification base on CNN and Transformer.☆24Updated 9 months ago
- SPA: Self-Peripheral-Attention for Central-Peripheral Interactions in Endoscopic Image Classification and Segmentation☆9Updated 11 months ago
- Code for UniS-MMC: Multimodal Classification via Unimodality-supervised Multimodal Contrastive Learning (ACL 2023)☆32Updated 7 months ago
- 这是一个clip-pytorch的模型,可以训练自己的数据集。☆206Updated last year
- DA-TransUNet: Combining Dual Attention of Position and Channel with Transformer U-net for Medical Image Segmentation☆141Updated 3 months ago
- 使用pytorch复现vit模型(图像分类)☆25Updated 2 years ago
- ☆49Updated last year
- Source code of the paper Multi-Granularity Part Sampling Attention for Fine-Grained Visual Classification☆22Updated 2 weeks ago
- The official code of "Rethinking Local Perception in Lightweight Vision Transformer"☆85Updated last year
- (MICCAI23) This is the official code repository for "EGE-UNet: an Efficient Group Enhanced UNet for skin lesion segmentation".☆259Updated last year
- ☆91Updated 10 months ago
- [IEEE TCYB 2024] CTNet: Contrastive Transformer Network for Polyp Segmentation☆23Updated last month
- [IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".☆78Updated this week
- ☆32Updated last year
- (BIBM22) This is the official code repository for "MALUNet: A Muti-Attention and Light-weight UNet for Skin Lesion Segmentation".☆73Updated last year
- pytorch resnet image-classification☆18Updated 2 years ago
- Code for Affine-Consistent Transformer for Multi-Class Cell Nuclei Detection (ICCV 2023)☆38Updated last year
- Pytorch Implement of UPerNet☆24Updated 2 years ago
- ☆32Updated 5 months ago
- 可视化特征图教程☆66Updated 2 years ago
- 细粒度图像分类之十二猫分类,对比ResNet和ViT两者模型性能。☆36Updated 2 years ago
- The public project of paper:"TongueSAM: An Universal Tongue Segmentation Model Based on SAM with Zero-Shot"☆28Updated last year
- SCSA: Exploring the Synergistic Effects Between Spatial and Channel Attention.☆60Updated 6 months ago
- An attention-based Multi-Scale Feature Learning Framework for Multimodal Medical Image Fusion☆28Updated 3 months ago
- [MICCAI'23] Official implementation of "RCS-YOLO: A Fast and High-Accuracy Object Detector for Brain Tumor Detection".☆82Updated this week
- Official code repository for: FCN-Transformer Feature Fusion for Polyp Segmention (MIUA 2022 paper)☆70Updated last year