用于pytorch的图像分类,包含多种模型方法,比如AlexNet,VGG,GoogleNet,ResNet,DenseNet等等,包含可完整运行的代码。除此之外,也有colab的在线运行代码,可以直接在colab在线运行查看结果。也可以迁移到自己的数据集进行迁移学习。
☆249Dec 7, 2023Updated 2 years ago
Alternatives and similar repositories for Pytorch-Image-Classification
Users that are interested in Pytorch-Image-Classification are comparing it to the libraries listed below
Sorting:
- 使用双层全连接神经网络和ResNet网络实现的图像二分类问题☆13Aug 1, 2020Updated 5 years ago
- Implementing LRP (Layer-wise Relevance Propagation) for a sequence-to-sequence model with GRU layers.☆12Sep 8, 2023Updated 2 years ago
- ☆10Sep 28, 2019Updated 6 years ago
- 基于PyTorch框架实现的图像分类网络☆85Nov 8, 2020Updated 5 years ago
- Multi-Source Domain Attention☆14Oct 21, 2020Updated 5 years ago
- PyTorch框架构建花朵图像分类模型(Resnet网络,迁移学习)☆19Aug 4, 2022Updated 3 years ago
- ☆17Dec 6, 2021Updated 4 years ago
- 实现常用图像分类算法☆46Jul 6, 2023Updated 2 years ago
- spatio-temporal tasks☆16Jul 15, 2024Updated last year
- Anomaly detection in maritime navigation with evidential deep learning based on graph-based clustering☆23Jul 27, 2021Updated 4 years ago
- Code for the paper: "U2Net: A General Framework with Spatial-Spectral-Integrated Double U-Net for Image Fusion", ACM MM 2023☆22Nov 14, 2023Updated 2 years ago
- Vessel traffic data, or Automatic Identification System (AIS) data, are collected by the U.S. Coast Guard through an onboard navigation s…☆23Nov 3, 2019Updated 6 years ago
- 基于keras集成多种图像分类模型: VGG16、VGG19、InceptionV3、Xception、MobileNet、AlexNet、LeNet、ZF_Net、ResNet18、ResNet34、ResNet50、ResNet_101、ResNet_152、Dense…☆474Nov 21, 2022Updated 3 years ago
- 论文“时变转速下基于改进图注意力网络的轴承半监督故障诊断”源码☆29May 9, 2022Updated 3 years ago
- Classification for cifar10 in pytorch including alexnet, densenet, googlenet, lenet, resnet, vgg☆26Dec 28, 2017Updated 8 years ago
- 基于卷积神经网络的遥感图像分类☆74Jul 27, 2025Updated 7 months ago
- ☆38Nov 8, 2024Updated last year
- Lightweight Rail Surface Defect Detection Algorithm Based on an Improved YOLOv8☆43Feb 27, 2025Updated last year
- ☆35Feb 10, 2023Updated 3 years ago
- 用CGAN训练时频域特征生成特征样本,数据集非图片数据集,用到了dataset自定义的数据集。☆12Apr 30, 2022Updated 3 years ago
- A CNN-BiLSTM model for Li-ion battery state of health and remaining useful life prediction☆11Mar 25, 2024Updated last year
- Implementation of Materials Discovery with Extreme properties via AI-Driven Combinatorial Chemistry☆10May 8, 2024Updated last year
- GAN Step By Step -- GSBS,顾名思义,我希望我自己能够一步一步的学习GAN。GAN 又名 生成对抗网络,是最近几年很热门的一种无监督算法,他能生成出非常逼真的照片,图像甚至视频。GAN是一个图像的全新的领域,从2014的GAN的发展现在,在计算机视觉中…☆11Jan 11, 2023Updated 3 years ago
- 图像二分类问题 猫狗大战 pytorch CNN☆138Jan 24, 2019Updated 7 years ago
- 利用pytorch实现图像分类的一个完整的代码 ,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆1,456Feb 6, 2023Updated 3 years ago
- MAE for CIFAR,由于可用资源有限,我们仅在 cifar10 上测试模型。我们主要想重现这样的结果:使用 MAE 预训练 ViT 可以比直接使用标签进行监督学习训练获得更好的结果。这应该是自我监督学习比监督学习更有效的数据的证据。☆81May 12, 2023Updated 2 years ago
- code for☆11Apr 10, 2021Updated 4 years ago
- A method for detecting anomalies consisting of unusual combinations of normal elements using set features☆36Nov 10, 2024Updated last year
- This repository contain the code we used to divide NinaPro database 5 into train set and test set☆11Mar 14, 2019Updated 6 years ago
- Code for the paper: "T-shape data and probabilistic remaining useful life prediction for Li-ion batteries using multiple non-crossing qua…☆10Aug 4, 2023Updated 2 years ago
- 一款要求读者支付比特币后才能阅读的WordPress付费阅读插件。A wordpress plugin, reader should pay bitcoin for reading.☆10Jun 22, 2014Updated 11 years ago
- Using the CGCNN transfer learning model to pridict the voltages of many kinds of metal-ion battery electrodes☆11Aug 23, 2022Updated 3 years ago
- moire detection using Haar transform☆10Dec 20, 2019Updated 6 years ago
- Deep Learning Project for Trajectory Prediction using nuScenes dataset.☆10Sep 13, 2022Updated 3 years ago
- ☆11Aug 8, 2018Updated 7 years ago
- Metal Parts’ Zero-Shot 6D Pose Estimation via Foundation Model and Template Update for Industrial Scenario☆17Feb 25, 2025Updated last year
- Vehicle object detection & distance estimation using thermal imaging.☆40Sep 3, 2022Updated 3 years ago
- ☆12Sep 28, 2024Updated last year
- [TGRS 2024 ESI Highly Cited Paper (TOP 1%)] Sliding Dual-Window-Inspired Reconstruction Network for Hyperspectral Anomaly Detection☆13Feb 28, 2024Updated 2 years ago