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()