Index

A

Accuracy (class in evaluator)

accuracy() (evaluator.Accuracy method) angle (dataloader.transform.RandomRotation attribute)

B

batch_size (dataloader.dataset.DataLoader attribute) build_dataloader() (in module dataloader.build) build_dataset() (in module dataloader.build) build_evaluator() (in module model.build)

build_feature_extractor() (in module model.build) build_feature_predictor() (in module model.build) build_model() (in module model.build) build_pretrained_model() (in module model.build) build_transform() (in module dataloader.build)

C

Compose (class in dataloader.transform) compute() (evaluator.Accuracy method) (evaluator.F1Score method) (evaluator.Metric method) (evaluator.Precision method) (evaluator.Recall method)

compute_visual_words() (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor method) CustomDataset (class in dataloader.dataset)

D

data_dir (dataloader.dataset.CustomDataset attribute) DataLoader (class in dataloader.dataset) dataloader.build module dataloader.dataset module

dataloader.transform module dataset (dataloader.dataset.DataLoader attribute) detector (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor attribute)

E

evaluate() (evaluator.Evaluator method) (models.Model method) evaluator module

Evaluator (class in evaluator) ExtractorPredictor (class in models)

F

F1Score (class in evaluator) f1score() (evaluator.F1Score method) FeatureExtractor (class in model.extractor_predictor.feature_extractor)

fit() (model.extractor_predictor.feature_extractor.FeatureExtractor method) (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor method) (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor method)

G

gamma (dataloader.transform.GammaCorrection attribute)

GammaCorrection (class in dataloader.transform) GrayScale (class in dataloader.transform)

I

image_paths (dataloader.dataset.CustomDataset attribute)

indices (dataloader.dataset.DataLoader attribute) input_shape (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor attribute)

L

launch module load_model() (model.extractor_predictor.feature_extractor.FeatureExtractor method) (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor class method) (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor class method) (models.ExtractorPredictor class method) (models.Model class method)

M

macro_accuracy() (evaluator.Accuracy method) macro_f1score() (evaluator.F1Score method) macro_precision() (evaluator.Precision method) macro_recall() (evaluator.Recall method) mean (dataloader.transform.Normalize attribute) Metric (class in evaluator) metrics (evaluator.Evaluator attribute) micro_accuracy() (evaluator.Accuracy method) micro_f1score() (evaluator.F1Score method) micro_precision() (evaluator.Precision method) micro_recall() (evaluator.Recall method) Model (class in models) model (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor attribute) model.build module

model.extractor_predictor.feature_extractor module models module module dataloader.build dataloader.dataset dataloader.transform evaluator launch model.build model.extractor_predictor.feature_extractor models scripts.test scripts.train scripts.visualization

N

n_visual_words (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor attribute)

Normalize (class in dataloader.transform)

P

pooling (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor attribute) Precision (class in evaluator) precision() (evaluator.Precision method) predict() (model.extractor_predictor.feature_extractor.FeatureExtractor method) (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor method) (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor method) (models.ExtractorPredictor method)

probability (dataloader.transform.RandomFlip attribute) (dataloader.transform.RandomRotation attribute) process_args() (in module launch)

R

RandomFlip (class in dataloader.transform) RandomRotation (class in dataloader.transform) Recall (class in evaluator)

recall() (evaluator.Recall method) reduce (evaluator.Recall attribute) Resize (class in dataloader.transform)

S

save_model() (models.ExtractorPredictor method) (models.Model method) scripts.test module scripts.train module scripts.visualization module serialize() (model.extractor_predictor.feature_extractor.FeatureExtractor method) (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor method) (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor method)

shuffle (dataloader.dataset.DataLoader attribute) SIFTFeatureExtractor (class in model.extractor_predictor.feature_extractor) size (dataloader.transform.Resize attribute) start_test() (in module scripts.test) start_train() (in module scripts.train) start_visualize() (in module scripts.visualization) std (dataloader.transform.Normalize attribute)

T

test() (in module scripts.test) train() (in module scripts.train) (models.ExtractorPredictor method) (models.Model method)

Transform (class in dataloader.transform) transform (dataloader.dataset.CustomDataset attribute) transform() (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor method) transforms (dataloader.transform.Compose attribute)

V

VGG16FeatureExtractor (class in model.extractor_predictor.feature_extractor)

visual_words (model.extractor_predictor.feature_extractor.SIFTFeatureExtractor attribute) visualize() (in module scripts.visualization)

W

weighted_accuracy() (evaluator.Accuracy method) weighted_f1score() (evaluator.F1Score method)

weighted_precision() (evaluator.Precision method) weighted_recall() (evaluator.Recall method) weights (model.extractor_predictor.feature_extractor.VGG16FeatureExtractor attribute)