diff options
Diffstat (limited to 'macros/trainImageCategoryClassifier.sci')
| -rw-r--r-- | macros/trainImageCategoryClassifier.sci | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/macros/trainImageCategoryClassifier.sci b/macros/trainImageCategoryClassifier.sci new file mode 100644 index 0000000..46ccce8 --- /dev/null +++ b/macros/trainImageCategoryClassifier.sci @@ -0,0 +1,31 @@ +function classifier = trainImageCategoryClassifier(imgSets, bag) +// This function is used to train an image classifier. +// +// Calling Sequence +// classifier = trainImageCategoryClassifier(imgSets, bag) +// +// Parameters +// classifier: Image category classifier +// imgSets: Input imageSet to train the classifier on +// bag: The bagOfFeatures of the imageSet provided +// +// Description +// This function trains an image category classifier which can be used to predict categories of images given to it as input using the predict() function. +// +// Examples +// imgSet = imageSet(directory,'recursive'); +// [trainingSet testSet] = partition(imgSet,[0.8]); +// bag = bagOfFeatures(trainingSet); +// categoryClassifier = trainImageCategoryClassifier(trainingSet, bag); +// +// Authors +// Rohit Suri +// Umang Agrawal + bag_list = bagStructToList(bag); + + imgSets_list = imageSetToList(imgSets); + temp = opencv_trainImageCategoryClassifier(imgSets_list, bag_list); + + classifier = struct("ClassifierLocation", temp(2), "BagofFeaturesLocation", temp(3), "Description", temp(4)) + +endfunction |