Skip to content

BDI-UFRGS/ContextFeatures

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Multi Scale Context Features for Image Classification

Method based on multi scale context features for image classification. This method allows the learning process to take advantage of all the information contained in the imagens, decreasing the information loss in the learning process. Our approach use the idea of aggregating image features at different scales, taking advantage of tranfer learning, being a suitable strategy to deal with images of different sizes and small datasets.

Code files

-context_features.py

This is the code of our approach. You can use it to generate all context features and concatenate them to create the multiscale context features. We also apply the train and test of the classifier in this code. Use -h tag to see the help of the code.

-simple_features.py

In this code, we use the approach based on simple features. We also apply the train and test of the classifier in this code.

-raw_image.py

This code uses the raw_image and train/test the model without transfer learning.

All the codes have a dictionary of the models, enable to use the same that we apply in the paper experiments.

These codes specifically use the Stanford Dataset that can be downloaded in: https://www.kaggle.com/datasets/jessicali9530/stanford-cars-dataset or http://ai.stanford.edu/~jkrause/cars/car_dataset.html. We use three different .mat files to get the labels and the classes names: 'cars_test_annos_withlabels_eval.mat', 'cars_train_annos.mat' and 'cars_annos.mat'. Make sure you have the three files and the images in a 'stanford/' folder.

To use CLIP you need to download and install it from: https://github.com/openai/CLIP.

Reference

For more information about the approach: link

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages