Skip to content

rafutek/CNN-active-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

76 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CNN Active Learning

This code allows to study active learning methods on several convolutional neural networks, with diferent image datasets, and other parameters, at once.

At the end of the experiment it shows the results in one or several plots depending on user arguments, and save them.

Therefore you can play with the results after, without relaunching all the experiment, that can be pretty long.

Active learning methods: least confidence sampling, margin sampling, and entropy sampling.

Convolutional Neural Networks: VggNet, ResNet and AlexNet.

Image datasets: CIFAR10 and CIFAR100.

How to use this code?

Open a terminal, and:

  • git clone https://github.com/rafutek/CNN-active-learning.git
  • cd CNN-active-learning
  • pip install -r requirements.txt
  • cd code
  • python experiment.py -h to display the help

Read the wiki for more informations


Special thanks to our professor Pierre-Marc Jodoin for its teaching, without him this project would not exist.

About

Study active learning methods on several CNNs and datasets.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •