From 2c7221e56c4b21dda5671497e4308bd2aea8b76c Mon Sep 17 00:00:00 2001 From: InnopolisU Date: Tue, 17 Dec 2024 17:41:34 +0300 Subject: [PATCH] Fixed gdcm error --- innofw/core/datasets/coco_rtk.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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):