根据分割图生成bbox的csv标注：

根据分割图生成bbox的csv标注：

import numpy as np
import skimage
from skimage.measure import label,regionprops
from skimage.filters import roberts
import os
import cv2
import pandas as pd

def get_patch_csv(filename_list, root_dir, output_path, area_thr = 9):
    # take the pred mask of segmention 
    annotation = pd.DataFrame()
    name_list = []
    x1_list = []
    y1_list = []
    x2_list = []
    y2_list = []
    bbox_id_list = []
    Bounding_boxes_list = []
    for filename in filename_list:
        im_path = os.path.join(root_dir, filename)
        img = cv2.imread(im_path)
        if len(img.shape)>2:
            img = img[:,:,0]
        label_img = label(img)
        bbox_id = 0
        for r in regionprops(label_img):
            if r.area < area_thr:
                continue
            name_list.append(filename[0:-3] + 'png')
            bbox_id_list.append(bbox_id)
            bbox_id += 1
            x1_list.append(r.bbox[1])
            y1_list.append(r.bbox[0])
            x2_list.append(r.bbox[3])
            y2_list.append(r.bbox[2])
            Bounding_boxes_list.append(','.join([str(r.bbox[i]) for i in [1,0,3,2]]))
    annotation['File_name'] = name_list
    annotation['bbox_id'] = bbox_id_list
    annotation['bbox_x1'] = x1_list
    annotation['bbox_y1'] = y1_list
    annotation['bbox_x2'] = x2_list
    annotation['bbox_y2'] = y2_list
    annotation['Bounding_boxes'] = Bounding_boxes_list
    annotation.to_csv(output_path)

if __name__ == "__main__":
    root_dir = ''
    output_path = 'xxx/seg_info.csv'
    filename_list = os.listdir(root_dir)
    get_patch_csv(filename_list, root_dir, output_path)