基于Python实现RLE格式分割标注文件的格式转换

落叶有声

1.Airbus Ship Detection Challenge

url: https://www.kaggle.com/competitions/airbus-ship-detection
Find ships on satellite images as quickly as possible
Data Description
In this competition, you are required to locate ships in images, and put an aligned bounding box segment around the ships you locate. Many images do not contain ships, and those that do may contain multiple ships. Ships within and across images may differ in size (sometimes significantly) and be located in open sea, at docks, marinas, etc.
For this metric, object segments cannot overlap. There were a small percentage of images in both the Train and Test set that had slight overlap of object segments when ships were directly next to each other. Any segments overlaps were removed by setting them to background (i.e., non-ship) encoding. Therefore, some images have a ground truth may be an aligned bounding box with some pixels removed from an edge of the segment. These small adjustments will have a minimal impact on scoring, since the scoring evaluates over increasing overlap thresholds.
The train_ship_segmentations.csv file provides the ground truth (in run-length encoding format) for the training images. The sample_submission files contains the images in the test images.
Please click on each file / folder in the Data Sources section to get more information about the files.

1. kaggle competitions download -c airbus-ship-detection

复制代码

2.数据展示


2.1 标注数据

该数据以csv格式存储，具体如下：

2.2 图象文件

3.格式转换

由于图太多，暂时转换10个

1. #!/usr/bin/env python3
   
2. # -*- coding: utf-8 -*-
   
3. 
   
4. import numpy as np  # linear algebra
   
5. import pandas as pd  # data processing, CSV file I/O (e.g. pd.read_csv)
   
6. from PIL import Image
   
7. 
   
8. 
   
9. # ref: https://www.kaggle.com/paulorzp/run-length-encode-and-decode
   
10. # 将图片编码成rle格式
    
11. def rle_encode(img, min_max_threshold=1e-3, max_mean_threshold=None):
    
12.     '''
    
13.     img: numpy array, 1 - mask, 0 - background
    
14.     Returns run length as string formated
    
15.     '''
    
16.     if np.max(img) < min_max_threshold:
    
17.         return ''  ## no need to encode if it's all zeros
    
18.     if max_mean_threshold and np.mean(img) > max_mean_threshold:
    
19.         return ''  ## ignore overfilled mask
    
20.     pixels = img.T.flatten()
    
21.     pixels = np.concatenate([[0], pixels, [0]])
    
22.     runs = np.where(pixels[1:] != pixels[:-1])[0] + 1
    
23.     runs[1::2] -= runs[::2]
    
24.     return ' '.join(str(x) for x in runs)
    
25. 
    
26. 
    
27. # 将图片从rle解码
    
28. def rle_decode(mask_rle, shape=(768, 768)):
    
29.     '''
    
30.     mask_rle: run-length as string formated (start length)
    
31.     shape: (height,width) of array to return
    
32.     Returns numpy array, 1 - mask, 0 - background
    
33.     '''
    
34.     s = mask_rle.split()
    
35.     starts, lengths = [np.asarray(x, dtype=int) for x in (s[0:][::2], s[1:][::2])]
    
36.     starts -= 1
    
37.     ends = starts + lengths
    
38.     img = np.zeros(shape[0] * shape[1], dtype=np.uint8)
    
39.     for lo, hi in zip(starts, ends):
    
40.         # img[lo:hi] = 1
    
41.         img[lo:hi] = 255 #方便可视化
    
42.     return img.reshape(shape).T  # Needed to align to RLE direction
    
43. 
    
44. 
    
45. def masks_as_image(in_mask_list):
    
46.     # Take the individual ship masks and create a single mask array for all ships
    
47.     all_masks = np.zeros((768, 768), dtype=np.uint8)
    
48.     for mask in in_mask_list:
    
49.         if isinstance(mask, str):
    
50.             all_masks |= rle_decode(mask)
    
51.     return all_masks
    
52. 
    
53. 
    
54. # 将目标路径下的rle文件中所包含的所有rle编码，保存到save_img_dir中去
    
55. def rle_2_img(train_rle_dir, save_img_dir):
    
56.     masks = pd.read_csv(train_rle_dir)
    
57.     not_empty = pd.notna(masks.EncodedPixels)
    
58.     print(not_empty.sum(), 'masks in', masks[not_empty].ImageId.nunique(), 'images')
    
59.     print((~not_empty).sum(), 'empty images in', masks.ImageId.nunique(), 'total images')
    
60.     all_batchs = list(masks.groupby('ImageId'))
    
61.     train_images = []
    
62.     train_masks = []
    
63.     i = 0
    
64.     for img_id, mask in all_batchs[:10]:
    
65.         c_mask = masks_as_image(mask['EncodedPixels'].values)
    
66.         im = Image.fromarray(c_mask)
    
67.         im.save(save_img_dir + img_id.split('.')[0] + '.png')
    
68.         print(i, img_id.split('.')[0] + '.png')
    
69.         i += 1
    
70. 
    
71.     return train_images, train_masks
    
72. 
    
73. 
    
74. if __name__ == '__main__':
    
75.     rle_2_img('train_ship_segmentations_v2.csv',
    
76.               'mask/')

复制代码

其中为了方便查看，原计划0为背景，1为mask，为了方便显示，设置为255为mask。

4.转换结果

到此这篇关于基于Python实现RLE格式分割标注文件的格式转换的文章就介绍到这了,更多相关Python RLE文件内容请搜索晓枫资讯以前的文章或继续浏览下面的相关文章希望大家以后多多支持晓枫资讯！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！