mmdetection跑通maskrcnn
https://github.com/open-mmlab/mmdetection
mmdetection说明文档
原格式为只包含polypoints的四个顶点,转化为用于训练mask rcnn的coco格式数据集。
"""
https://github.com/wusaifei/prepare_detection_dataset/blob/master/labelme2coco.py
"""
import os
import json
import numpy as np
import cv2
classname_to_id = {
"card": 1}
def order_points_clock(pts):
pts = np.array(pts).reshape((-1, 2))
xSorted = pts[np.argsort(pts[:, 0]), :]
leftMost = xSorted[:2, :]
rightMost = xSorted[2:, :]
leftMost = leftMost[np.argsort(leftMost[:, 1]), :]
(tl, bl) = leftMost
rightMost = rightMost[np.argsort(rightMost[:, 1]), :]
(tr, br) = rightMost
return np.array([tl, tr, br, bl], dtype=np.int32)
class MyEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, np.integer):
return int(obj)
elif isinstance(obj, np.floating)