fix: script for miou calculation

Author: fracape

compressai_vision/codecs/sic_sfu2022.py

@@ -38,6 +38,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+
 from torch.hub import load_state_dict_from_url
 
 from compressai.ans import BufferedRansEncoder, RansDecoder

compressai_vision/datasets/image.py

@@ -37,6 +37,7 @@
 from typing import Dict, List
 
 import numpy as np
+
 from detectron2.data import DatasetCatalog, MetadataCatalog
 from detectron2.data.common import DatasetFromList, MapDataset
 from detectron2.data.dataset_mapper import DatasetMapper

compressai_vision/model_wrappers/detectron2.py

@@ -27,6 +27,7 @@
 # ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 import re
+
 from enum import Enum
 from pathlib import Path
 from typing import Dict, List
@@ -81,7 +82,6 @@ def __init__(self, *args, **kwargs) -> None:
         self.activation = activation
 
     def set_attributes(self, module):
-
         if hasattr(module, "norm"):
             self.norm = module.norm
 
@@ -125,7 +125,6 @@ def __init__(self, *args, **kwargs) -> None:
         self.activation = activation
 
     def set_attributes(self, module):
-
         if hasattr(module, "norm"):
             self.norm = module.norm
 

compressai_vision/model_wrappers/intconv2d.py

@@ -32,6 +32,7 @@
 
 import numpy as np
 import torch
+
 from torch.nn import functional as F
 
 

compressai_vision/model_wrappers/jde_lowlevel.py

@@ -29,6 +29,7 @@
 
 import torch
 import torch.nn as nn
+
 from jde.models import EmptyLayer, Upsample, YOLOLayer
 
 from .intconv2d import IntConv2d

compressai_vision/pipelines/remote_inference/video_remote_inference.py

@@ -198,7 +198,9 @@ def __call__(
             )  # Assuming one qp will be used
 
             if not isinstance(res["bitstream"], dict):
-                out_res["bytes"] = Path(res["bitstream"]).stat().st_size / len(dataloader)
+                out_res["bytes"] = Path(res["bitstream"]).stat().st_size / len(
+                    dataloader
+                )
             else:
                 assert len(res["bytes"]) == len(dataloader)
                 out_res["bytes"] = res["bytes"][e]

scripts/metrics/compute_overall_miou.py

@@ -33,6 +33,7 @@
 
 
 """
+
 from __future__ import annotations
 
 import argparse
@@ -43,6 +44,8 @@
 
 from compressai_vision.evaluators.evaluators import BaseEvaluator
 
+from .compute_overall_mota import compute_overall_mota
+
 CLASSES = ["PANDAM1", "PANDAM2", "PANDAM2"]
 
 SEQS_BY_CLASS = {
@@ -92,7 +95,6 @@
 
 
 def compute_overall_mIoU(class_name, items):
-
     miou_acc = 0.0
     for item in items:
         with open(item["eval_info"], "r") as f: