diff --git a/innofw/core/datasets/coco_rtk.py b/innofw/core/datasets/coco_rtk.py index b3be1136..78c74b36 100644 --- a/innofw/core/datasets/coco_rtk.py +++ b/innofw/core/datasets/coco_rtk.py @@ -1,13 +1,17 @@ +from pckg_util import install_and_import + +install_and_import("pylibjpeg", "2.0.0", packageimportname="pylibjpeg") +install_and_import("python-gdcm", "3.0.24.1", packageimportname="gdcm") +from innofw.utils.data_utils.preprocessing.dicom_handler import dicom_to_raster import json import os - import cv2 import numpy as np import pydicom import torch from torch.utils.data import Dataset -from innofw.utils.data_utils.preprocessing.dicom_handler import dicom_to_raster + class DicomCocoDatasetRTK(Dataset):