We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d026e26 commit 3511b87Copy full SHA for 3511b87
compressai_vision/datasets/image.py
@@ -31,11 +31,13 @@
31
import base64
32
import logging
33
import re
34
+
35
from glob import glob
36
from pathlib import Path
37
from typing import Dict, List
38
39
import numpy as np
40
41
from detectron2.data import DatasetCatalog, MetadataCatalog
42
from detectron2.data.common import DatasetFromList, MapDataset
43
from detectron2.data.dataset_mapper import DatasetMapper
tests/evaluators/test_evaluators.py
@@ -29,11 +29,12 @@
29
30
import json
import os
-import pytest
-import torch
from unittest.mock import MagicMock, patch
+import pytest
+import torch
from compressai_vision.evaluators import COCOEVal
0 commit comments