This is the offical website for paper ''Category-consistent deep network learning for accurate vehicle logo recognition''

Overview

The Pytorch Implementation of Category-consistent deep network learning for accurate vehicle logo recognition

Framework Architecture

Image

Requirements

  • Pytorch==1.0.1 or higher
  • opencv version: 4.1.0

Datasets

  • XMU:
    • Y. Huang, R. Wu, Y. Sun, W. Wang, and X. Ding, Vehicle logo recog775 nition system based on convolutional neural networks with a pretraining strategy, IEEE Transactions on Intelligent Transportation Systems 16 (4) (2015) 1951-1960.
    • https://xmu-smartdsp.github.io/VehicleLogoRecognition.html
  • HFUT-VL1 and HFUT-VL2:
    • Y. Yu, J. Wang, J. Lu, Y. Xie, and Z. Nie, Vehicle logo recognition based on overlapping enhanced patterns of oriented edge magnitudes, Computers & Electrical Engineering 71 (2018) 273–283.
    • https://github.com/HFUT-VL/HFUT-VL-dataset
  • CompCars:
    • L. Yang, P. Luo, C. C. Loy, and X. Tang, A large-scale car dataset for fine-grained categorization and verification, in: Proc. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Boston, MA, 2015, pp. 3973-3981.
    • http://mmlab.ie.cuhk.edu.hk/datasets/comp_cars/index.html
  • VLD-45:

VLF-net for classification (Vehicle logo feature extraction network)

  • Training with the classification pipeline

    • training XMU dataset
    python train.py --dataset_name XMU --framework Classification_Network
    
    • training HFUT-VL1 dataset
    python train.py --dataset_name HFUT_VL1 --framework Classification_Network
    
    • training HFUT-VL2 dataset
    python train.py --dataset_name HFUT_VL2 --framework Classification_Network
    
    • training CompCars dataset
    python train.py --dataset_name CompCars --framework Classification_Network
    
    • training VLD-45 dataset
    python train.py --dataset_name VLD-45 --framework Classification_Network
    
  • Testing with the classification pipeline

    • testing XMU dataset
    python test.py --dataset_name XMU --framework Classification_Network
    
    • testing HFUT-VL1 dataset
    python test.py --dataset_name HFUT_VL1 --framework Classification_Network
    
    • testing HFUT-VL2 dataset
    python test.py --dataset_name HFUT_VL2 --framework Classification_Network
    
    • testing CompCars dataset
    python test.py --dataset_name CompCars --framework Classification_Network
    
    • testing VLD-45 dataset
    python test.py --dataset_name VLD-45 --framework Classification_Network
    

VLF-net for category-consistent mask learning

  • Step 1:

    • Generation of the category-consistent masks. There are more details for the co-localization method PSOL.
    • Please note that we use the generated binary-masks directly instead of the predicted boxes.
  • Step 2:

    • After generating the category-consistent masks, we can further organize the training and testing data which are as below:
    root/
          test/
              dog/xxx.png
              dog/xxz.png
              cat/123.png
              cat/nsdf3.png
          train/
              dog/xxx.png
              dog/xxz.png
              cat/123.png
              cat/nsdf3.png
          mask/
              dog/xxx.png
              dog/xxz.png
              cat/123.png
              cat/nsdf3.png
    
    Note that each image has the corresponding generated category-consistent mask.
  • Step 3:

    • Now, you can training the model with the category-consistent mask learning framework

    • Training with the category-consistent deep network learning framework pipeline

      • training XMU dataset
      python train.py --dataset_name XMU --framework CCML_Network
      
      • training HFUT-VL1 dataset
      python train.py --dataset_name HFUT_VL1 --framework CCML_Network
      
      • training HFUT-VL2 dataset
      python train.py --dataset_name HFUT_VL2 --framework CCML_Network
      
      • training CompCars dataset
      python train.py --dataset_name CompCars --framework CCML_Network
      
      • training VLD-45 dataset
      python train.py --dataset_name VLD-45 --framework CCML_Network
      
    • Testing with the category-consistent deep network learning framework pipeline

      • testing XMU dataset
      python test.py --dataset_name XMU --framework CCML_Network
      
      • testing HFUT-VL1 dataset
      python test.py --dataset_name HFUT_VL1 --framework CCML_Network
      
      • testing HFUT-VL2 dataset
      python test.py --dataset_name HFUT_VL2 --framework CCML_Network
      
      • testing CompCars dataset
      python test.py --dataset_name CompCars --framework CCML_Network
      
      • testing VLD-45 dataset
      python test.py --dataset_name VLD-45 --framework CCML_Network
      

Experiments

Image

Image

Bibtex

  • If you find our code useful, please cite our paper:
    @article{LU2021,
    title = {Category-consistent deep network learning for accurate vehicle logo recognition},
      journal = {Neurocomputing},
      year = {2021},
      issn = {0925-2312},
      doi = {https://doi.org/10.1016/j.neucom.2021.08.030},
      url = {https://www.sciencedirect.com/science/article/pii/S0925231221012145},
      author = {Wanglong Lu and Hanli Zhao and Qi He and Hui Huang and Xiaogang Jin}
      }
    

Acknowledgements

Owner
Wanglong Lu
I am a Ph.D. student at Ubiquitous Computing and Machine Learning Research Lab (UCML), Memorial University of Newfoundland.
Wanglong Lu
QuakeLabeler is a Python package to create and manage your seismic training data, processes, and visualization in a single place — so you can focus on building the next big thing.

QuakeLabeler Quake Labeler was born from the need for seismologists and developers who are not AI specialists to easily, quickly, and independently bu

Hao Mai 15 Nov 04, 2022
Deep generative modeling for time-stamped heterogeneous data, enabling high-fidelity models for a large variety of spatio-temporal domains.

Neural Spatio-Temporal Point Processes [arxiv] Ricky T. Q. Chen, Brandon Amos, Maximilian Nickel Abstract. We propose a new class of parameterizations

Facebook Research 75 Dec 19, 2022
Data and code for ICCV 2021 paper Distant Supervision for Scene Graph Generation.

Distant Supervision for Scene Graph Generation Data and code for ICCV 2021 paper Distant Supervision for Scene Graph Generation. Introduction The pape

THUNLP 23 Dec 31, 2022
JugLab 33 Dec 30, 2022
Lolviz - A simple Python data-structure visualization tool for lists of lists, lists, dictionaries; primarily for use in Jupyter notebooks / presentations

lolviz By Terence Parr. See Explained.ai for more stuff. A very nice looking javascript lolviz port with improvements by Adnan M.Sagar. A simple Pytho

Terence Parr 785 Dec 30, 2022
Sharing of contents on mitochondrial encounter networks

mito-network-sharing Sharing of contents on mitochondrial encounter networks Required: R with igraph, brainGraph, ggplot2, and XML libraries; igraph l

Stochastic Biology Group 0 Oct 01, 2021
PICARD - Parsing Incrementally for Constrained Auto-Regressive Decoding from Language Models

This is the official implementation of the following paper: Torsten Scholak, Nathan Schucher, Dzmitry Bahdanau. PICARD - Parsing Incrementally for Con

ElementAI 217 Jan 01, 2023
This is a repository for a Semantic Segmentation inference API using the Gluoncv CV toolkit

BMW Semantic Segmentation GPU/CPU Inference API This is a repository for a Semantic Segmentation inference API using the Gluoncv CV toolkit. The train

BMW TechOffice MUNICH 56 Nov 24, 2022
On Out-of-distribution Detection with Energy-based Models

On Out-of-distribution Detection with Energy-based Models This repository contains the code for the experiments conducted in the paper On Out-of-distr

Sven 19 Aug 07, 2022
MicroNet: Improving Image Recognition with Extremely Low FLOPs (ICCV 2021)

MicroNet: Improving Image Recognition with Extremely Low FLOPs (ICCV 2021) A pytorch implementation of MicroNet. If you use this code in your research

Yunsheng Li 293 Dec 28, 2022
Fast Differentiable Matrix Sqrt Root

Official Pytorch implementation of ICLR 22 paper Fast Differentiable Matrix Square Root

YueSong 42 Dec 30, 2022
Aspect-Sentiment-Multiple-Opinion Triplet Extraction (NLPCC 2021)

The code and data for the paper "Aspect-Sentiment-Multiple-Opinion Triplet Extraction" Requirements Python 3.6.8 torch==1.2.0 pytorch-transformers==1.

慢半拍 5 Jul 02, 2022
Research code for the paper "How Good is Your Tokenizer? On the Monolingual Performance of Multilingual Language Models"

Introduction This repository contains research code for the ACL 2021 paper "How Good is Your Tokenizer? On the Monolingual Performance of Multilingual

AdapterHub 20 Aug 04, 2022
Face Detection and Alignment using Multi-task Cascaded Convolutional Networks (MTCNN)

Face-Detection-with-MTCNN Face detection is a computer vision problem that involves finding faces in photos. It is a trivial problem for humans to sol

Chetan Hirapara 3 Oct 07, 2022
A baseline code for VSPW

A baseline code for VSPW Preparation Download VSPW dataset The VSPW dataset with extracted frames and masks is available here.

28 Aug 22, 2022
Image-retrieval-baseline - MUGE Multimodal Retrieval Baseline

MUGE Multimodal Retrieval Baseline This repo is implemented based on the open_cl

47 Dec 16, 2022
Laplace Redux -- Effortless Bayesian Deep Learning

Laplace Redux - Effortless Bayesian Deep Learning This repository contains the code to run the experiments for the paper Laplace Redux - Effortless Ba

Runa Eschenhagen 28 Dec 07, 2022
Inflated i3d network with inception backbone, weights transfered from tensorflow

I3D models transfered from Tensorflow to PyTorch This repo contains several scripts that allow to transfer the weights from the tensorflow implementat

Yana 479 Dec 08, 2022
Prototypical Cross-Attention Networks for Multiple Object Tracking and Segmentation, NeurIPS 2021 Spotlight

PCAN for Multiple Object Tracking and Segmentation This is the offical implementation of paper PCAN for MOTS. We also present a trailer that consists

ETH VIS Group 328 Dec 29, 2022
A GUI to automatically create a TOPAS-readable MLC simulation file

Python script to create a TOPAS-readable simulation file descriring a Multi-Leaf-Collimator. Builds the MLC using the data from a 3D .stl file.

Sebastian Schäfer 0 Jun 19, 2022