Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -1,6 +1,49 @@
|
|
1 |
"""
|
2 |
Implementation of YOLOv3 architecture
|
3 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
from pytorch_lightning import LightningModule
|
5 |
import torch
|
6 |
from loss import YoloLoss
|
|
|
1 |
"""
|
2 |
Implementation of YOLOv3 architecture
|
3 |
"""
|
4 |
+
import os
|
5 |
+
import pytorch_lightning as pl
|
6 |
+
import pandas as pd
|
7 |
+
import seaborn as sn
|
8 |
+
import torch
|
9 |
+
import torch.nn as nn
|
10 |
+
import torch.nn.functional as F
|
11 |
+
import torchvision
|
12 |
+
from IPython.core.display import display
|
13 |
+
#from pl_bolts.datamodules import CIFAR10DataModule
|
14 |
+
#from pl_bolts.transforms.dataset_normalizations import cifar10_normalization
|
15 |
+
from pytorch_lightning import LightningModule, Trainer, seed_everything
|
16 |
+
from pytorch_lightning.callbacks import LearningRateMonitor, ModelCheckpoint
|
17 |
+
from pytorch_lightning.callbacks.progress import TQDMProgressBar
|
18 |
+
from pytorch_lightning.loggers import CSVLogger
|
19 |
+
from torch.optim.lr_scheduler import OneCycleLR
|
20 |
+
from torchmetrics.functional import accuracy
|
21 |
+
import torch.cuda.amp as amp
|
22 |
+
from torch.utils.data import DataLoader
|
23 |
+
from loss import YoloLoss
|
24 |
+
from pytorch_lightning import LightningModule, Trainer
|
25 |
+
from torch.optim.lr_scheduler import OneCycleLR
|
26 |
+
from torch_lr_finder import LRFinder
|
27 |
+
import torch.nn as nn
|
28 |
+
from dataset import YOLODataset
|
29 |
+
import config
|
30 |
+
import torch
|
31 |
+
import torch.optim as optim
|
32 |
+
import os
|
33 |
+
from model import YOLOv3
|
34 |
+
from tqdm import tqdm
|
35 |
+
from utils import (
|
36 |
+
mean_average_precision,
|
37 |
+
cells_to_bboxes,
|
38 |
+
get_evaluation_bboxes,
|
39 |
+
save_checkpoint,
|
40 |
+
load_checkpoint,
|
41 |
+
check_class_accuracy,
|
42 |
+
get_loaders,
|
43 |
+
plot_couple_examples
|
44 |
+
)
|
45 |
+
from loss import YoloLoss
|
46 |
+
import warnings
|
47 |
from pytorch_lightning import LightningModule
|
48 |
import torch
|
49 |
from loss import YoloLoss
|