Welcome to Manufacturing Part Classification’s documentation!

Contents:

• Manufacturing-Part-Classification
  • LAUNCH
    • process_args()
  • MODEL
    • Evaluator
      • Accuracy
        • Accuracy.accuracy()
        • Accuracy.compute()
        • Accuracy.macro_accuracy()
        • Accuracy.micro_accuracy()
        • Accuracy.weighted_accuracy()
      • Evaluator
        • Evaluator.metrics
        • Evaluator.evaluate()
      • F1Score
        • F1Score.compute()
        • F1Score.f1score()
        • F1Score.macro_f1score()
        • F1Score.micro_f1score()
        • F1Score.weighted_f1score()
      • Metric
        • Metric.compute()
      • Precision
        • Precision.compute()
        • Precision.macro_precision()
        • Precision.micro_precision()
        • Precision.precision()
        • Precision.weighted_precision()
      • Recall
        • Recall.reduce
        • Recall.compute()
        • Recall.macro_recall()
        • Recall.micro_recall()
        • Recall.recall()
        • Recall.weighted_recall()
    • Models Zoo
      • ExtractorPredictor
        • ExtractorPredictor.load_model()
        • ExtractorPredictor.predict()
        • ExtractorPredictor.save_model()
        • ExtractorPredictor.train()
      • Model
        • Model.evaluate()
        • Model.load_model()
        • Model.save_model()
        • Model.train()
    • Model Builder
      • build_evaluator()
      • build_feature_extractor()
      • build_feature_predictor()
      • build_model()
      • build_pretrained_model()
    • ExtractorPredictor
      • FeatureExtractor
        • FeatureExtractor.fit()
        • FeatureExtractor.load_model()
        • FeatureExtractor.predict()
        • FeatureExtractor.serialize()
      • SIFTFeatureExtractor
        • SIFTFeatureExtractor.visual_words
        • SIFTFeatureExtractor.n_visual_words
        • SIFTFeatureExtractor.detector
        • SIFTFeatureExtractor.compute_visual_words()
        • SIFTFeatureExtractor.fit()
        • SIFTFeatureExtractor.load_model()
        • SIFTFeatureExtractor.predict()
        • SIFTFeatureExtractor.serialize()
        • SIFTFeatureExtractor.transform()
      • VGG16FeatureExtractor
        • VGG16FeatureExtractor.model
        • VGG16FeatureExtractor.input_shape
        • VGG16FeatureExtractor.weights
        • VGG16FeatureExtractor.pooling
        • VGG16FeatureExtractor.fit()
        • VGG16FeatureExtractor.load_model()
        • VGG16FeatureExtractor.predict()
        • VGG16FeatureExtractor.serialize()
  • SCRIPTS
    • start_train()
    • train()
    • start_test()
    • test()
    • start_visualize()
    • visualize()
  • DATALOADER
    • Dataset
      • CustomDataset
        • CustomDataset.data_dir
        • CustomDataset.transform
        • CustomDataset.image_paths
      • DataLoader
        • DataLoader.dataset
        • DataLoader.batch_size
        • DataLoader.shuffle
        • DataLoader.indices
    • Transformations
      • Compose
        • Compose.transforms
      • GammaCorrection
        • GammaCorrection.gamma
      • GrayScale
      • Normalize
        • Normalize.mean
        • Normalize.std
      • RandomFlip
        • RandomFlip.probability
      • RandomRotation
        • RandomRotation.probability
        • RandomRotation.angle
      • Resize
        • Resize.size
      • Transform
    • Dataloader Builder
      • build_dataloader()
      • build_dataset()
      • build_transform()

Indices and tables

• Index

• Module Index

• Search Page