YINYIPENG-EN / vit_classification_pytorch
采用vit实现图像分类
☆21Updated 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的基础之上,将前三层特征提取网络作为不同输入的特征提取网络,在中间层进行特征拼接,后面的卷积层用于提取融合特征,最后加上全连接层。该网络稍作修改就能同时提取…☆84Updated 4 years ago
- 可视化神经网络☆39Updated 2 years ago
- 基于Swin-transformer训练图像分类并部署web端☆85Updated 2 years ago
- 这是一个clip-pytorch的模型,可以训练自己的数据集。☆211Updated last year
- 使用pytorch完成的一个多模态分类任务,文本和图像部分分别使用了bert和resnet提取特征(在config里可以组合多种模型),在我的小规模数据集上取得了良好的性能(验证集acc96%)☆72Updated last year
- 使用pytorch复现vit模型(图像分类)☆27Updated 2 years ago
- An attention-based Multi-Scale Feature Learning Framework for Multimodal Medical Image Fusion☆30Updated last week
- The public project of paper:"TongueSAM: An Universal Tongue Segmentation Model Based on SAM with Zero-Shot"☆32Updated 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☆144Updated 4 months ago
- ☆50Updated last year
- This repo is about multi-label medical image classification base on CNN and Transformer.☆24Updated 10 months ago
- HiFuse: Hierarchical Multi-Scale Feature Fusion Network for Medical Image Classification☆216Updated last month
- SPA: Self-Peripheral-Attention for Central-Peripheral Interactions in Endoscopic Image Classification and Segmentation☆10Updated last year
- (BIBM22) This is the official code repository for "MALUNet: A Muti-Attention and Light-weight UNet for Skin Lesion Segmentation".☆80Updated last year
- The official PyTorch implementation of our paper "MCA: Multidimensional collaborative attention in deep convolutional neural networks for…☆57Updated last year
- 商品图像检索、多模态、深度学习☆31Updated 3 years ago
- The official code of "Rethinking Local Perception in Lightweight Vision Transformer"☆85Updated last year
- Source code of the paper Multi-Granularity Part Sampling Attention for Fine-Grained Visual Classification☆23Updated last month
- [IEEE TIP 2022] Official implementation of MATR: Multimodal Medical Image Fusion via Multiscale Adaptive Transformer☆80Updated 11 months ago
- ☆75Updated 6 months ago
- deep learning for image processing including classification and object-detection etc.☆24Updated 3 years ago
- ☆48Updated 6 months ago
- ☆32Updated 6 months ago
- mamba在Windows的环境配置☆22Updated 10 months ago
- Official code repository for: FCN-Transformer Feature Fusion for Polyp Segmention (MIUA 2022 paper)☆70Updated last year
- ☆34Updated last year
- Code for UniS-MMC: Multimodal Classification via Unimodality-supervised Multimodal Contrastive Learning (ACL 2023)☆33Updated 9 months ago
- [IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".☆81Updated last month
- ICONIP 2024☆91Updated last week