Update model.py

model.py

@@ -1,6 +1,49 @@
 """
 Implementation of YOLOv3 architecture
 """
+import os
+import pytorch_lightning as pl
+import pandas as pd
+import seaborn as sn
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+import torchvision
+from IPython.core.display import display
+#from pl_bolts.datamodules import CIFAR10DataModule
+#from pl_bolts.transforms.dataset_normalizations import cifar10_normalization
+from pytorch_lightning import LightningModule, Trainer, seed_everything
+from pytorch_lightning.callbacks import LearningRateMonitor, ModelCheckpoint
+from pytorch_lightning.callbacks.progress import TQDMProgressBar
+from pytorch_lightning.loggers import CSVLogger
+from torch.optim.lr_scheduler import OneCycleLR
+from torchmetrics.functional import accuracy
+import torch.cuda.amp as amp
+from torch.utils.data import DataLoader
+from loss import YoloLoss
+from pytorch_lightning import LightningModule, Trainer
+from torch.optim.lr_scheduler import OneCycleLR
+from torch_lr_finder import LRFinder
+import torch.nn as nn
+from dataset import YOLODataset
+import config
+import torch
+import torch.optim as optim
+import os
+from model import YOLOv3
+from tqdm import tqdm
+from utils import (
+    mean_average_precision,
+    cells_to_bboxes,
+    get_evaluation_bboxes,
+    save_checkpoint,
+    load_checkpoint,
+    check_class_accuracy,
+    get_loaders,
+    plot_couple_examples
+)
+from loss import YoloLoss
+import warnings
 from pytorch_lightning import LightningModule
 import torch
 from loss import YoloLoss