deepml
- deepml package
- Subpackages
- Submodules
- deepml.accelerator_trainer module
- deepml.base module
BaseLearnerBaseLearner.__init__()BaseLearner.set_optimizer()BaseLearner.set_criterion()BaseLearner.set_lr_scheduler_policy()BaseLearner.load_optimizer_state()BaseLearner.load_lr_schedular_state()BaseLearner.create_state_dict()BaseLearner.save()BaseLearner.init_metrics()BaseLearner.update_metrics()BaseLearner.update_metrics_with_simple_moving_average()BaseLearner.write_metrics_to_logger()BaseLearner.write_lr()BaseLearner.log_metrics()BaseLearner.fit()BaseLearner.predict()
- deepml.constants module
- deepml.datasets module
ImageRowDataFrameDatasetImageDataFrameDatasetImageDataFrameDataset.dataframeImageDataFrameDataset.image_file_name_columnImageDataFrameDataset.target_columnsImageDataFrameDataset.image_dirImageDataFrameDataset.transformsImageDataFrameDataset.samplesImageDataFrameDataset.target_transformImageDataFrameDataset.open_file_funcImageDataFrameDataset.__init__()ImageDataFrameDataset.__len__()ImageDataFrameDataset.__getitem__()
ImageListDatasetSegmentationDataFrameDatasetSegmentationDataFrameDataset.dataframeSegmentationDataFrameDataset.image_dirSegmentationDataFrameDataset.mask_dirSegmentationDataFrameDataset.image_colSegmentationDataFrameDataset.mask_colSegmentationDataFrameDataset.albu_torch_transformsSegmentationDataFrameDataset.target_transformSegmentationDataFrameDataset.samplesSegmentationDataFrameDataset.trainSegmentationDataFrameDataset.open_file_funcSegmentationDataFrameDataset.__init__()SegmentationDataFrameDataset.__len__()SegmentationDataFrameDataset.__getitem__()
- deepml.fabric_trainer module
- deepml.losses module
- deepml.lr_scheduler_utils module
- deepml.tasks module
TaskTask.__init__()Task.modelTask.model_dirTask.deviceTask.model_file_nameTask.move_input_to_device()Task.transform_input()Task.transform_target()Task.transform_output()Task.predict_batch()Task.train_step()Task.eval_step()Task.predict()Task.predict_class()Task.show_predictions()Task.write_prediction_to_logger()Task.evaluate()
NeuralNetTaskNeuralNetTask.__init__()NeuralNetTask.predict_batch()NeuralNetTask.train_step()NeuralNetTask.eval_step()NeuralNetTask.predict()NeuralNetTask.predict_class()NeuralNetTask.show_predictions()NeuralNetTask.transform_target()NeuralNetTask.transform_output()NeuralNetTask.write_prediction_to_logger()NeuralNetTask.evaluate()
SegmentationSegmentation.modeSegmentation.num_classesSegmentation.thresholdSegmentation.class_index_to_colorSegmentation.paletteSegmentation.__init__()Segmentation.predict_batch()Segmentation.save_prediction()Segmentation.predict_class()Segmentation.show_predictions()Segmentation.transform_target()Segmentation.transform_output()Segmentation.decode_segmentation_mask()Segmentation.log_prediction()Segmentation.write_prediction_to_logger()
ImageRegressionImageClassificationMultiLabelImageClassification
- deepml.tracking module
- deepml.trainer module
LearnerLearner.epochs_completedLearner.best_val_lossLearner.historyLearner.loggerLearner.__init__()Learner.set_optimizer()Learner.set_criterion()Learner.set_lr_scheduler()Learner.save()Learner.validate()Learner.set_predictor()Learner.fit()Learner.predict()Learner.predict_class()Learner.extract_features()Learner.show_predictions()
- deepml.transforms module
- deepml.utils module
- deepml.visualize module
- Module contents