ошибка при попытке загрузить данные

Question

Я пытался проанализировать мой набор данных с изображениями и аннотациями с кодом ниже.Я использую Anaconda с IDE Spyder на компьютере с Windows:

DATA_DIR = 'input'
# Directory to save logs and trained model
ROOT_DIR = 'working'
train_dicom_dir = os.path.join(DATA_DIR, 'stage_1_train_images')
test_dicom_dir = os.path.join(DATA_DIR, 'stage_1_test_images')
print(train_dicom_dir)
print(test_dicom_dir)
def get_dicom_fps(dicom_dir):
    dicom_fps = glob.glob(dicom_dir+'/'+'*.jpg')
    return list(set(dicom_fps))
def parse_dataset(dicom_dir, anns): 
    image_fps = get_dicom_fps(dicom_dir)
    image_annotations = {fp: [] for fp in image_fps}
    for index, row in anns.iterrows(): 
        fp = os.path.join(dicom_dir, row['patientId']+'.jpg')
        image_annotations[fp].append(row)
    return image_fps, image_annotations

image_fps, image_annotations = parse_dataset(train_dicom_dir, anns=anns)

При запуске кода я получаю следующую ошибку:

Traceback (most recent call last):
  File "<ipython-input-21-fe2564bb2360>", line 1, in <module>
    image_fps, image_annotations = parse_dataset(train_dicom_dir, anns=anns)
  File "<ipython-input-12-a5d26437db38>", line 6, in parse_dataset
    image_annotations[fp].append(row)
KeyError: 'input\\stage_1_train_images\\1.2.276.0.7230010.3.1.4.8323329.10001.1517874346.163716.jpg'

Мой текущий рабочий каталог - 'C: \Пользователи \ rajaramans2 \ кодов \ вход '

Answer 1

Похоже, вы пытаетесь обработать изображения в формате .dcm (формат DICOM), а не .jpg, поэтому ваш код должен работать, если вы замените «.jpg» на «.dcm», надеюсь, это поможет.