PRIS-CV/Mutual-Channel-Loss PRIS-CV/Mutual-Channel-Loss

  • Stars
    star
    255
  • Rank 159,729 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created almost 5 years ago
  • Updated over 2 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
PRIS-CV/Mutual-Channel-Loss
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 The Devil is in the Channels: Mutual-Channel Loss for Fine-Grained Image Classification (TIP 2020)

The Devil is in the Channels: Mutual-Channel Loss for Fine-Grained Image Classification

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

Changelog

  • 2020/09/14 update the code: CUB-200-2011_ResNet18.py Training with ResNet18 (TRAINED FROM SCRATCH).
  • 2020/04/19 add the hyper-parameter fine-tune results.
  • 2020/04/18 clean the code for better understanding.

Dataset

CUB-200-2011

Requirements

  • python 3.6
  • PyTorch 1.2.0
  • torchvision

Training

  • Download datasets
  • Train: python CUB-200-2011.py, the alpha and beta are the hyper-parameters of the MC-Loss
  • Description : PyTorch CUB-200-2011 Training with VGG16 (TRAINED FROM SCRATCH).

Hyper-parameter

Loss = ce_loss + alpha_1 * L_dis + beta_1 * L_div
Hyper-parameter_1 Hyper-parameter_2 The figure is plot by NNI.

Other versions

Other unofficial implements can be found in the following:

  • Kurumi233: This repo integrate the MC-Loss into a class. code
  • darcula1993: This repo implement the tf version of the MC-Loss. code
  • Holocron: Implementations of recent Deep Learning tricks in Computer Vision, easily paired up with your favorite framework and model zoo. code

Citation

If you find this paper useful in your research, please consider citing:

@ARTICLE{9005389, 
author={D. {Chang} and Y. {Ding} and J. {Xie} and A. K. {Bhunia} and X. {Li} and Z. {Ma} and M. {Wu} and J. {Guo} and Y. {Song}}, 
journal={IEEE Transactions on Image Processing}, 
title={The Devil is in the Channels: Mutual-Channel Loss for Fine-Grained Image Classification}, 
year={2020}, volume={29}, number={}, pages={4683-4695}, 
doi={10.1109/TIP.2020.2973812}, 
ISSN={1941-0042}, 
month={},}

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

PMG-Progressive-Multi-Granularity-Training

Code release for Fine-Grained Visual Classification via Progressive Multi-Granularity Training of Jigsaw Patches (ECCV2020)
Python
216
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