Update

Co-Authored-By: Ranuga-Disansa <[email protected]>

Author: Dewanai1

Model/__init__.py

@@ -0,0 +1,52 @@
+# Imports
+import gc
+import threading
+import ast
+import os
+import urllib.request
+import cv2
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+import torch
+import wandb
+from detectron2 import model_zoo
+from detectron2.config import get_cfg
+from detectron2.config.config import CfgNode
+from detectron2.data import DatasetCatalog, MetadataCatalog, build_detection_test_loader
+from detectron2.engine import DefaultPredictor, DefaultTrainer
+from detectron2.evaluation import COCOEvaluator, inference_on_dataset
+from detectron2.structures import BoxMode
+from detectron2.utils.logger import setup_logger
+from detectron2.utils.visualizer import Visualizer
+from sklearn.model_selection import ParameterGrid
+from torchmetrics import (
+    PSNR,
+    SSIM,
+    MeanAbsoluteError,
+    MeanSquaredError,
+    Precision,
+    Recall,
+)
+
+try:
+    from tqdm import tqdm
+except Exception as e:
+    raise ImportError(
+        f"""
+        Cannot Import Tqdm try installing it using 
+        `pip3 install tqdm` 
+        or 
+        `conda install tqdm`.
+        \n 
+        {e}"""
+    )
+try:
+    from ray import tune
+except:
+    tune = None
+
+from Model.dataset import *
+from Model.help_funcs import *
+from Model.metrics import *
+from Model.modelling import *

Model/dataset/__init__.py

@@ -0,0 +1 @@
+from Model.dataset.download import *