PRIS-CV/PMG-Progressive-Multi-Granularity-Training PRIS-CV/PMG-Progressive-Multi-Granularity-Training

  • Stars
    star
    216
  • Rank 183,179 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created over 4 years ago
  • Updated over 3 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
PRIS-CV/PMG-Progressive-Multi-Granularity-Training
github

PRIS-CV

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Code release for Fine-Grained Visual Classification via Progressive Multi-Granularity Training of Jigsaw Patches (ECCV2020)

Progressive Multi-Granularity Training

Code release for Fine-Grained Visual Classification via Progressive Multi-Granularity Training of Jigsaw Patches (ECCV2020)

Requirement

python 3.6

PyTorch >= 1.3.1

torchvision >= 0.4.2

Training

  1. Download datatsets for FGVC (e.g. CUB-200-2011, Standford Cars, FGVC-Aircraft, etc) and organize the structure as follows:
dataset
β”œβ”€β”€ train
β”‚   β”œβ”€β”€ class_001
|   |      β”œβ”€β”€ 1.jpg
|   |      β”œβ”€β”€ 2.jpg
|   |      └── ...
β”‚   β”œβ”€β”€ class_002
|   |      β”œβ”€β”€ 1.jpg
|   |      β”œβ”€β”€ 2.jpg
|   |      └── ...
β”‚   └── ...
└── test
    β”œβ”€β”€ class_001
    |      β”œβ”€β”€ 1.jpg
    |      β”œβ”€β”€ 2.jpg
    |      └── ...
    β”œβ”€β”€ class_002
    |      β”œβ”€β”€ 1.jpg
    |      β”œβ”€β”€ 2.jpg
    |      └── ...
    └── ...
  1. Train from scratch with train.py.

Citation

Please cite our paper if you use PMG in your work.

@InProceedings{du2020fine,
  title={Fine-Grained Visual Classification via Progressive Multi-Granularity Training of Jigsaw Patches},
  author={Du, Ruoyi and Chang, Dongliang and Bhunia, Ayan Kumar and Xie, Jiyang and Song, Yi-Zhe and Ma, Zhanyu and Guo, Jun},
  booktitle = {European Conference on Computer Vision},
  year={2020}
}

Contact

Thanks for your attention! If you have any suggestion or question, you can leave a message here or contact us directly:

More Repositories

1

DemoFusion

Let us democratise high-resolution generation! (CVPR 2024)
Jupyter Notebook
1,961
star
2

Mutual-Channel-Loss

Code release for The Devil is in the Channels: Mutual-Channel Loss for Fine-Grained Image Classification (TIP 2020)
Python
255
star
3

Fine-Grained-or-Not

Code release for Your β€œFlamingo” is My β€œBird”: Fine-Grained, or Not (CVPR 2021 Oral)
Python
57
star
4

On-the-fly-Category-Discovery

Code release for Your β€œOn-the-fly Category Discovery (CVPR 2023)”
Python
49
star
5

BSNet

Code release for the paper BSNet: Bi-Similarity Network for Few-shot Fine-grained Image Classification. (TIP2020)
Python
49
star
6

Bi-FRN

Code release for Bi-Directional Feature Reconstruction Network for Fine-grained Few-shot Image Classification
Python
48
star
7

AP-CNN_Pytorch-master

Weakly Supervised Attention Pyramid Convolutional Neural Network for Fine-Grained Visual Classification (TIP2021)
Python
47
star
8

OSLNet

Code release for OSLNet: Deep Small-Sample Classification with an Orthogonal Softmax Layer (TIP2020)
Python
44
star
9

DCRNet

The repository contains the PyTorch implementation of "Duplex Contextual Relations for PolypSegmentation"
Python
34
star
10

seal

Semantic Enhanced Attribute Learning
Jupyter Notebook
34
star
11

WFEN

[ACMMM 2024] "Efficient Face Super-Resolution via Wavelet-based Feature Enhancement Network"
Python
32
star
12

Fine-grained-Visual-Analysis-Library

FGVCLib is an open-source and well documented library for Fine-grained Visual Classification.
Python
31
star
13

PCA-Net

Progressive Co-Attention Network for Fine-Grained Visual Classification
Python
31
star
14

DSACA

Code release for Dilated-Scale-Aware Category-Attention ConvNet for Multi-Class Object Counting
Python
19
star
15

Mutual-to-Separate

Code release for Mind the Gap: Enlarging the Domain Gap in Open Set Domain Adaptation (TCSVT 2023)
Python
18
star
16

AdvancedDropout

Advanced Dropout: A Model-free Methodology for Bayesian Dropout Optimization (IEEE TPAMI 2021)
Python
15
star
17

PMG-V2

Code release for "Progressive Learning of Category-Consistent Multi-Granularity Features for Fine-Grained Visual Classification"
Python
15
star
18

An-Erudite-FGVC-Model

Code release for Your β€œAn Erudite Fine-Grained Visual Classification Model (CVPR 2023)"
Python
14
star
19

Making-a-Bird-AI-Expert-Work-for-You-and-Me

Code release for "Making a Bird AI Expert Work for You and Me (TPAMI 2023)".
Python
14
star
20

NeRSP

Python
13
star
21

Top-Down-Spatial-Attention-Loss

Fine-Grained Visual Classification via Simultaneously Learning of Multi-regional Multi-grained Features
Python
12
star
22

CAM-Guided-Attention

Code release for Grad-CAM Guided Attention Module for Fine-grained Visual Classification (MLSP 2022)
Python
12
star
23

MSSRM

An implementation of MSSRM method
Python
11
star
24

RelMatch

Code release for "Clue Me In: Semi-Supervised FGVC with Out-of-Distribution Data".
Python
11
star
25

IVR

Python
9
star
26

FFDI

Python
9
star
27

knowledge-transfer-based-FGVC

knowledge transfer based fine-grained visual classification
Python
9
star
28

Pascal-EA

Python
9
star
29

Category-Specific-Prompt

Code release for "Category-Specific Prompts for Animal Action Recognition with Pretrained Vision-Language Models"
Python
9
star
30

BiEN

Code release for Bi-Directional Ensemble Network for Few-Shot Fine-Grained Classification.
Python
8
star
31

DropChannelBlock_Pytorch_master

Python
8
star
32

Semantic-Memory-Guided-Image-Representation-for-Polyp-Segmentation

Semantic Memory Guided Image Representation for Polyp Segmentation (ICASSP-2023)
Python
8
star
33

RTMem

Python
7
star
34

Sketch-SF

Python
6
star
35

CMF-Refseg

Code of our ICIP 2021 paper CMF: Cascaded Multi-model Fusion for Referring Image Segmentation
6
star
36

TA2-Net

Python
5
star
37

MAFR

Code release for "Multi-View Active Fine-Grained Visual Recognition" (ICCV 2023οΌ‰
Python
5
star
38

SSKD

Code release for SSKD: Self-Supervised Knowledge Distillation for Cross Domain Adaptive Person Re-Identification
Python
4
star
39

ReMarNet

Code release for the paper ReMarNet: Conjoint Relation and Margin Learning for Small-Sample Image Classification (TCSVT 2020)
Python
4
star
40

HumanRecon

Python
4
star
41

DL-CV-ITS

Code release for Deep Learning-based Computer Vision for Surveillance in ITS: Evaluation of State-of-the-art Methods (IEEE TVT 2021)
Python
4
star
42

Sketch-CS

Python
4
star
43

Pair-wise-Similarity-module

Python
3
star
44

Attribute-Comprehension-of-VLMs

We provide a benchmark for evaluating the attribute understanding capabilities of large vision-language models.
Python
3
star
45

CN-CNN

Cross-layer Navigation Convolutional Neural Network for Fine-grained Visual Classification
Python
3
star
46

CGVC

Cross-Layer Feature based Multi-Granularity Visual Classification
Python
3
star
47

InterBoost

Code release for Deep InterBoost Networks for Small-sample Image Classification (NEUROCOMPUTING 2020)
Python
3
star
48

GPCA

GPCA: A Probabilistic Framework for Gaussian Process Embedded Channel Attention (IEEE TPAMI 2021)
3
star
49

Mixture-of-Hand-Experts

Python
3
star
50

AFGR

2
star
51

DS-UI

DS-UI: Dual-Supervised Mixture of Gaussian Mixture Models for Uncertainty Inference in Image Recognition (IEEE TIP 2021)
Python
2
star
52

Caption-Feature-Space-Regularization

This is the code for "Caption Feature Space Regularization for Audio Captioning"
Python
2
star
53

EGNN_TLRM

Code release for the paper: "TLRM: Task-level Relation Module for GNN-based Few-Shot Learning" (IEEE VCIP 2021)
Python
2
star
54

ENDE_For_SSS

Python
2
star
55

class-level-sampling

Python
1
star
56

Structured-DropConnect

Structured DropConnect for Uncertainty Inference in Image Classification
Python
1
star
57

Adaptive-Multi-Resolution-Feature-Fusion

Python
1
star
58

Fine-Grained-Age-Estimation-in-the-Wild-With-Attention-LSTM-Networks

Lua
1
star
59

DFA

Dual-granularity Feature Alignment for Cross-modality Person Re-identification
Python
1
star
60

Reserve_to_Adapt

Python
1
star
61

CRL-code

Code release for the paper β€œCompeting Ratio Loss for Discriminative Multi-class Image Classification” (IEEE Neurocomputing 2021).
Python
1
star