Невозможно проверить, находится ли поле изображения в пределах изображения в python - PullRequest
Новая
       46

Невозможно проверить, находится ли поле изображения в пределах изображения в python

0 голосов
/ 13 апреля 2020

Изображения, которые я пытаюсь проверить, правильно ли размещен ограничивающий прямоугольник, выдают ошибку, сообщающую, что тип изображения - Нет. Изображения в формате jpg, ниже приведен код, который я запускаю. В сгенерированных мной файлах csv перечислены изображения в первой строке программы excel в формате cucumber_001.jpg.jpg, 864,482, cucumber, 402,176,421,300. Есть ли проблема с дублированием расширения jpg. Как я могу подойти к этому? Должен ли я удалить одно расширение jpg? При запуске кода 

images_path = 'images'
for CSV_FILE in ['train_labels2.csv', 'test_labels.csv']:
  with open(CSV_FILE, 'r') as fid:  
      print('[*] Checking file:', CSV_FILE) 
      file = csv.reader(fid, delimiter=',')
      first = True 
      cnt = 0
      error_cnt = 0
      error = False
      for row in file:
          if error == True:
              error_cnt += 1
              error = False         
          if first == True:
              first = False
              continue     
          cnt += 1      
          name, width, height, xmin, ymin, xmax, ymax = row[0], int(row[1]), int(row[2]), int(row[4]), int(row[5]), int(row[6]), int(row[7])     
          path = os.path.join(images_path, name)
          img = cv2.imread(path)         
          if type(img) == type(None):
              error = True
              print('Could not read image', img)
              continue     
          org_height, org_width = img.shape[:2]     
          if org_width != width:
              error = True
              print('Width mismatch for image: ', name, width, '!=', org_width)     
          if org_height != height:
              error = True
              print('Height mismatch for image: ', name, height, '!=', org_height) 
          if xmin > org_width:
              error = True
              print('XMIN > org_width for file', name)  
          if xmax > org_width:
              error = True
              print('XMAX > org_width for file', name)
          if ymin > org_height:
              error = True
              print('YMIN > org_height for file', name)
          if ymax > org_height:
              error = True
              print('YMAX > org_height for file', name)
          if error == True:
              print('Error for file: %s' % name)
              print()
      print()
      print('Checked %d files and realized %d errors' % (cnt, error_cnt))
      print("-----")```
появляется сообщение «Не удалось прочитать изображение»
...