Skip to content

Commit 2c7221e

Browse files
committed
Fixed gdcm error
1 parent fdc534e commit 2c7221e

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

innofw/core/datasets/coco_rtk.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,17 @@
1+
from pckg_util import install_and_import
2+
3+
install_and_import("pylibjpeg", "2.0.0", packageimportname="pylibjpeg")
4+
install_and_import("python-gdcm", "3.0.24.1", packageimportname="gdcm")
5+
from innofw.utils.data_utils.preprocessing.dicom_handler import dicom_to_raster
16
import json
27
import os
3-
48
import cv2
59
import numpy as np
610
import pydicom
711
import torch
812
from torch.utils.data import Dataset
913

10-
from innofw.utils.data_utils.preprocessing.dicom_handler import dicom_to_raster
14+
1115

1216

1317
class DicomCocoDatasetRTK(Dataset):

0 commit comments

Comments
 (0)