.gitignore
.pre-commit-config.yaml
CHANGELOG.md
LICENSE.md
README.md
cuda_dev_env.yml
pyproject.toml
setup.cfg
setup.py
.github/workflows/master.yml
.github/workflows/test.yml
.vscode/settings.json
examples/dvc/.gitignore
examples/dvc/DVC workflow.ipynb
examples/dvc/data/generated_samples/sample_1/preethi-viswanathan-h5fsy4E4FMY-unsplash.jpg
examples/dvc/data/generated_samples/sample_2/nasa-Q1p7bh3SHj8-unsplash.jpg
examples/dvc/data/generated_samples/sample_3/jj-ying-8bghKxNU1j0-unsplash.jpg
examples/function_to_config/function_to_config.ipynb
examples/function_to_config/settings.ini
examples/function_to_config/test_function_to_config/__init__.py
examples/function_to_config/test_function_to_config/_nbdev.py
examples/function_to_config/test_function_to_config/test_function_to_config.py
examples/image_bbox_detection/coco/.gitignore
examples/image_bbox_detection/coco/README.md
examples/image_bbox_detection/coco/end_to_end_torchvision_faster_rcnn.ipynb
examples/image_bbox_detection/coco/settings.ini
examples/image_bbox_detection/coco/configs/augmentation.yml
examples/image_bbox_detection/coco/configs/callbacks.yml
examples/image_bbox_detection/coco/configs/data_setup.yml
examples/image_bbox_detection/coco/configs/dataset.yml
examples/image_bbox_detection/coco/configs/loss.yml
examples/image_bbox_detection/coco/configs/metric.yml
examples/image_bbox_detection/coco/configs/model.yml
examples/image_bbox_detection/coco/configs/optimizer.yml
examples/image_bbox_detection/coco/configs/splitter.yml
examples/image_bbox_detection/coco/configs/trainer.yml
examples/image_bbox_detection/coco/configs/transformations.yml
examples/image_bbox_detection/coco/configs/variables.yml
examples/image_bbox_detection/coco/data/images/test2017/000000000061.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000009.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000025.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000030.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000034.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000036.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000042.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000049.jpg
examples/image_bbox_detection/coco/data/images/train2017/000000000061.jpg
examples/image_bbox_detection/coco/data/labels/test2017/000000000061.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000009.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000025.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000030.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000034.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000036.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000042.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000049.txt
examples/image_bbox_detection/coco/data/labels/train2017/000000000061.txt
examples/image_bbox_detection/coco/test_end_to_end_image_bounding_box_detection/__init__.py
examples/image_bbox_detection/coco/test_end_to_end_image_bounding_box_detection/_nbdev.py
examples/image_bbox_detection/coco/test_end_to_end_image_bounding_box_detection/test_end_to_end_torchvision_faster_rcnn.py
examples/image_classification/.gitignore
examples/image_classification/README.md
examples/image_classification/resnet_image_classification_mlflow_model.ipynb
examples/image_classification/settings.ini
examples/image_classification/test_end_to_end_resnet_image_classification.ipynb
examples/image_classification/configs/augmentation.yml
examples/image_classification/configs/callbacks.yml
examples/image_classification/configs/data_setup.yml
examples/image_classification/configs/dataset.yml
examples/image_classification/configs/loss.yml
examples/image_classification/configs/model.yml
examples/image_classification/configs/optimizer.yml
examples/image_classification/configs/splitter.yml
examples/image_classification/configs/trainer.yml
examples/image_classification/configs/transformations.yml
examples/image_classification/configs/variables.yml
examples/image_classification/data/hotdog_not_hotdog_test/test/hot_dog/133012.jpg
examples/image_classification/data/hotdog_not_hotdog_test/test/not_hot_dog/6229.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/hot_dog/2417.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/hot_dog/3690.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/hot_dog/5079.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/hot_dog/76722.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/hot_dog/7896.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/not_hot_dog/1164.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/not_hot_dog/1167.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/not_hot_dog/197.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/not_hot_dog/4176.jpg
examples/image_classification/data/hotdog_not_hotdog_test/train/not_hot_dog/4770.jpg
examples/image_classification/models/model_for_test/checkpoints/epoch=0-step=2.ckpt.config.yml
examples/image_classification/models/model_for_test/mlflow/MLmodel
examples/image_classification/models/model_for_test/mlflow/conda.yaml
examples/image_classification/models/model_for_test/mlflow/python_model.pkl
examples/image_classification/models/model_for_test/tensorboard/events.out.tfevents.1640068458.poop-eating-sashimi.29366.0
examples/image_classification/models/model_for_test/tensorboard/Phase.TEST_All Losses_Phase.TEST/CrossEntropyLoss-0/events.out.tfevents.1640068467.poop-eating-sashimi.29366.8
examples/image_classification/models/model_for_test/tensorboard/Phase.TEST_All Losses_Total/events.out.tfevents.1640068467.poop-eating-sashimi.29366.7
examples/image_classification/models/model_for_test/tensorboard/Phase.TRAIN_All Losses_Phase.TRAIN/CrossEntropyLoss-0/events.out.tfevents.1640068466.poop-eating-sashimi.29366.3
examples/image_classification/models/model_for_test/tensorboard/Phase.TRAIN_All Losses_Total/events.out.tfevents.1640068466.poop-eating-sashimi.29366.2
examples/image_classification/models/model_for_test/tensorboard/Phase.VALIDATION_All Losses_Phase.VALIDATION/CrossEntropyLoss-0/events.out.tfevents.1640068466.poop-eating-sashimi.29366.5
examples/image_classification/models/model_for_test/tensorboard/Phase.VALIDATION_All Losses_Total/events.out.tfevents.1640068466.poop-eating-sashimi.29366.4
examples/image_classification/test_image_classification_end_to_end/__init__.py
examples/image_classification/test_image_classification_end_to_end/_nbdev.py
examples/image_classification/test_image_classification_end_to_end/mlflow_resnet_model_prediction.py
examples/image_classification/test_image_classification_end_to_end/test_end_to_end_resnet.py
examples/image_segmentation/.gitignore
examples/image_segmentation/Mask generation.ipynb
examples/image_segmentation/settings.ini
examples/image_segmentation/test_end_to_end_unet.ipynb
examples/image_segmentation/configs/augmentation.yml
examples/image_segmentation/configs/callbacks.yml
examples/image_segmentation/configs/data_setup.yml
examples/image_segmentation/configs/dataset.yml
examples/image_segmentation/configs/loss.yml
examples/image_segmentation/configs/model.yml
examples/image_segmentation/configs/optimizer.yml
examples/image_segmentation/configs/splitter.yml
examples/image_segmentation/configs/trainer.yml
examples/image_segmentation/configs/transformations.yml
examples/image_segmentation/configs/variables.yml
examples/image_segmentation/data/berlin_segmentation_test/test/images/berlin190_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/test/images/berlin191_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/test/masks/berlin190_image.png
examples/image_segmentation/data/berlin_segmentation_test/test/masks/berlin191_image.png
examples/image_segmentation/data/berlin_segmentation_test/train/images/berlin1_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/train/images/berlin2_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/train/images/berlin3_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/train/images/berlin4_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/train/images/berlin5_image.jpg
examples/image_segmentation/data/berlin_segmentation_test/train/masks/berlin1_image.png
examples/image_segmentation/data/berlin_segmentation_test/train/masks/berlin2_image.png
examples/image_segmentation/data/berlin_segmentation_test/train/masks/berlin3_image.png
examples/image_segmentation/data/berlin_segmentation_test/train/masks/berlin4_image.png
examples/image_segmentation/data/berlin_segmentation_test/train/masks/berlin5_image.png
examples/image_segmentation/test_image_segmentation_end_to_end/__init__.py
examples/image_segmentation/test_image_segmentation_end_to_end/_nbdev.py
examples/image_segmentation/test_image_segmentation_end_to_end/test_end_to_end_unet.py
examples/mean_average_precision/Mean Average Precision Example.ipynb
examples/mean_average_precision/annotations.csv
examples/mean_average_precision/detections.csv
requirements/generate_requirements.py
requirements/requirements.py
roheboam/__init__.py
roheboam/cli.py
roheboam.egg-info/PKG-INFO
roheboam.egg-info/SOURCES.txt
roheboam.egg-info/dependency_links.txt
roheboam.egg-info/entry_points.txt
roheboam.egg-info/not-zip-safe
roheboam.egg-info/requires.txt
roheboam.egg-info/top_level.txt
roheboam/engine/.gitignore
roheboam/engine/README.md
roheboam/engine/__init__.py
roheboam/engine/logger.py
roheboam/engine/core/__init__.py
roheboam/engine/core/callbacks/__init__.py
roheboam/engine/core/callbacks/callback.py
roheboam/engine/core/callbacks/handler.py
roheboam/engine/core/callbacks/recorders/__init__.py
roheboam/engine/core/callbacks/recorders/callback_method_is_called_recorder.py
roheboam/engine/core/callbacks/recorders/image_bounding_box_detection_recorder.py
roheboam/engine/core/callbacks/recorders/image_classification_recorder.py
roheboam/engine/core/callbacks/recorders/image_segmentation_recorder.py
roheboam/engine/core/callbacks/recorders/loss_recorder.py
roheboam/engine/core/callbacks/recorders/tensorboard_recorder.py
roheboam/engine/core/callbacks/recorders/utils.py
roheboam/engine/core/callbacks/savers/__init__.py
roheboam/engine/core/callbacks/savers/mlflow_saver.py
roheboam/engine/core/callbacks/savers/model_saver.py
roheboam/engine/core/data/__init__.py
roheboam/engine/core/data/data.py
roheboam/engine/core/losses/__init__.py
roheboam/engine/core/losses/core.py
roheboam/engine/core/losses/focal_loss.py
roheboam/engine/core/losses/mock_loss.py
roheboam/engine/core/losses/torchvision_fasterrcnn_loss.py
roheboam/engine/core/losses/yolo_loss.py
roheboam/engine/core/metrics/__init__.py
roheboam/engine/core/metrics/box_iou.py
roheboam/engine/core/metrics/core.py
roheboam/engine/core/metrics/mean_average_precision/__init__.py
roheboam/engine/core/metrics/mean_average_precision/label_statistics.py
roheboam/engine/core/metrics/mean_average_precision/mean_average_precision.py
roheboam/engine/core/optimizer/__init__.py
roheboam/engine/core/optimizer/adam.py
roheboam/engine/core/optimizer/ranger.py
roheboam/engine/core/optimizer/sgd.py
roheboam/engine/core/scheduler/__init__.py
roheboam/engine/core/scheduler/cosine_delayed.py
roheboam/engine/core/training/__init__.py
roheboam/engine/core/training/data/__init__.py
roheboam/engine/core/training/data/settings.ini
roheboam/engine/core/training/data/core/__init__.py
roheboam/engine/core/training/data/core/_nbdev.py
roheboam/engine/core/training/data/core/data_loader.py
roheboam/engine/core/training/data/notebooks/data_loader.ipynb
roheboam/engine/core/training/splitting/__init__.py
roheboam/engine/core/training/splitting/core.py
roheboam/engine/integrations/__init__.py
roheboam/engine/integrations/activeloop/__init__.py
roheboam/engine/integrations/activeloop/hub.py
roheboam/engine/integrations/activeloop/utils.py
roheboam/engine/integrations/docker/__init__.py
roheboam/engine/integrations/docker/containerise_model.py
roheboam/engine/integrations/docker/run_docker_container.py
roheboam/engine/integrations/docker/utils.py
roheboam/engine/integrations/dvc/__init__.py
roheboam/engine/integrations/mlflow/__init__.py
roheboam/engine/integrations/mlflow/loading.py
roheboam/engine/integrations/mlflow/saving.py
roheboam/engine/integrations/pytorch_lightning/__init__.py
roheboam/engine/integrations/pytorch_lightning/trainer.py
roheboam/engine/integrations/pytorch_lightning/adapters/__init__.py
roheboam/engine/integrations/pytorch_lightning/adapters/callback.py
roheboam/engine/integrations/pytorch_lightning/adapters/module.py
roheboam/engine/integrations/pytorch_lightning/callbacks/__init__.py
roheboam/engine/integrations/pytorch_lightning/callbacks/model_checkpoint.py
roheboam/engine/integrations/pytorch_lightning/callbacks/on_epoch_end_transformer.py
roheboam/engine/pipeline/README.md
roheboam/engine/pipeline/__init__.py
roheboam/engine/pipeline/config.py
roheboam/engine/pipeline/config_loader.py
roheboam/engine/pipeline/config_recorder.py
roheboam/engine/pipeline/constants.py
roheboam/engine/pipeline/graph_construction.py
roheboam/engine/pipeline/node.py
roheboam/engine/pipeline/pipeline.py
roheboam/engine/pipeline/yaml_loader.py
roheboam/engine/pipeline/utils/__init__.py
roheboam/engine/pipeline/utils/lookup_creation.py
roheboam/engine/pipeline/utils/lookup_functions.py
roheboam/engine/testing/__init__.py
roheboam/engine/testing/image_bounding_box_detection.py
roheboam/engine/utils/__init__.py
roheboam/engine/utils/conda.py
roheboam/engine/utils/convenience.py
roheboam/engine/utils/hooks.py
roheboam/engine/utils/interfaces.py
roheboam/engine/utils/model.py
roheboam/engine/utils/s3.py
roheboam/engine/utils/saving.py
roheboam/engine/utils/testing.py
roheboam/engine/utils/training.py
roheboam/engine/vision/__init__.py
roheboam/engine/vision/core/__init__.py
roheboam/engine/vision/core/augmentations/__init__.py
roheboam/engine/vision/core/augmentations/albumentation.py
roheboam/engine/vision/core/models/__init__.py
roheboam/engine/vision/core/models/cadene/__init__.py
roheboam/engine/vision/core/models/cadene/change_last_linear.py
roheboam/engine/vision/core/models/torchvision/__init__.py
roheboam/engine/vision/core/models/unet/__init__.py
roheboam/engine/vision/core/models/unet/core.py
roheboam/engine/vision/core/models/utils/__init__.py
roheboam/engine/vision/core/models/utils/layers.py
roheboam/engine/vision/tasks/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/dataset.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/sample/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/sample/create.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/sample/filter.py
roheboam/engine/vision/tasks/image_bounding_box_detection/data/sample/sample.py
roheboam/engine/vision/tasks/image_bounding_box_detection/deployment/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/deployment/mlflow.py
roheboam/engine/vision/tasks/image_bounding_box_detection/loaders/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/loaders/activeloop.py
roheboam/engine/vision/tasks/image_bounding_box_detection/loaders/roheboam.py
roheboam/engine/vision/tasks/image_bounding_box_detection/loaders/ultralytics.py
roheboam/engine/vision/tasks/image_bounding_box_detection/modules/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/modules/pytorch_lightning/__init__.py
roheboam/engine/vision/tasks/image_bounding_box_detection/modules/pytorch_lightning/torchvision_fasterrcnn.py
roheboam/engine/vision/tasks/image_bounding_box_detection/modules/pytorch_lightning/yolov4.py
roheboam/engine/vision/tasks/image_classification/__init__.py
roheboam/engine/vision/tasks/image_classification/data/__init__.py
roheboam/engine/vision/tasks/image_classification/data/dataset.py
roheboam/engine/vision/tasks/image_classification/data/sample.py
roheboam/engine/vision/tasks/image_classification/deployment/__init__.py
roheboam/engine/vision/tasks/image_classification/deployment/mlflow.py
roheboam/engine/vision/tasks/image_classification/loaders/__init__.py
roheboam/engine/vision/tasks/image_classification/loaders/activeloop.py
roheboam/engine/vision/tasks/image_classification/loaders/label_per_folder.py
roheboam/engine/vision/tasks/image_classification/modules/__init__.py
roheboam/engine/vision/tasks/image_classification/modules/pytorch_lightning/__init__.py
roheboam/engine/vision/tasks/image_classification/modules/pytorch_lightning/image_classification.py
roheboam/engine/vision/tasks/image_segmentation/__init__.py
roheboam/engine/vision/tasks/image_segmentation/data/__init__.py
roheboam/engine/vision/tasks/image_segmentation/data/dataset.py
roheboam/engine/vision/tasks/image_segmentation/data/sample.py
roheboam/engine/vision/tasks/image_segmentation/deployment/__init__.py
roheboam/engine/vision/tasks/image_segmentation/deployment/mlflow.py
roheboam/engine/vision/tasks/image_segmentation/loaders/__init__.py
roheboam/engine/vision/tasks/image_segmentation/loaders/activeloop.py
roheboam/engine/vision/tasks/image_segmentation/loaders/from_paths.py
roheboam/engine/vision/tasks/image_segmentation/modules/__init__.py
roheboam/engine/vision/tasks/image_segmentation/modules/pytorch_lightning/__init__.py
roheboam/engine/vision/tasks/image_segmentation/modules/pytorch_lightning/unet.py
roheboam/engine/vision/utils/__init__.py
roheboam/engine/vision/utils/bbox.py
roheboam/engine/vision/utils/normalization.py
roheboam/engine/vision/utils/rle.py
roheboam/engine/vision/utils/transforms.py
roheboam/engine/vision/utils/u_net_weight_map.py
roheboam/engine/vision/utils/image/__init__.py
roheboam/engine/vision/utils/image/io.py
roheboam/engine/vision/utils/image/plot.py
roheboam/engine/vision/utils/image/tile_with_square.py
roheboam/engine/vision/utils/image/tile_with_square_for_image_with_bboxes.py
tests/engine/callbacks/test_image_bounding_box_detection_recorder.py
tests/engine/callbacks/test_tensorboard_recorder.py
tests/engine/pipeline/test_config_loader.py
tests/engine/pipeline/test_config_recorder.py
tests/engine/pipeline/test_graph_construction.py
tests/engine/pipeline/test_pipeline_integration_for_nodes_to_be_removed.py
tests/engine/vision/tasks/image_bounding_box_detection/data/__init__.py
tests/engine/vision/tasks/image_bounding_box_detection/data/dataset/__init__.py
tests/engine/vision/tasks/image_bounding_box_detection/data/dataset/test_image_bounding_box_detection_torchvision_collate_fn.py
tests/engine/vision/tasks/image_bounding_box_detection/data/dataset/test_image_bounding_box_detection_torchvision_collate_fn_with_no_bboxes_or_labels.py
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/test_load_roheboam_sample.py
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000009/metadata.json
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000009/image/000000000009.jpg
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000009/label/000000000009.txt
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000025/metadata.json
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000025/image/000000000025.jpg
tests/engine/vision/tasks/image_bounding_box_detection/data/loaders/roheboam/data/000000000025/label/000000000025.txt
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/__init__.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_create_bbox_defection_sample.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_create_bbox_sample_with_different_num_of_bboxes_and_labels.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_create_bbox_sample_without_bbox_format.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_create_plot_array.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_filtering_of_labels_and_bboxes_for_bbox_detection_sample.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_lazy_filtering_of_labels_and_bboxes_within_sample.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_lazy_loading_of_bboxes_for_bbox_detection_sample.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_sample_has_label.py
tests/engine/vision/tasks/image_bounding_box_detection/data/sample/test_transform_label.py
tests/engine/vision/tasks/image_bounding_box_detection/modules/pytorch_lightning/test_torchvision_fasterrcnn.py
tests/engine/vision/utils/image/io/test_saving_and_loading_grayscale.py
tests/engine/vision/utils/image/io/test_data/berlin1_mask.png