Python调用MMDetection实现AI抠图去背景
coco_80_dict = {
'person': {'index': 0, 'id': 1, 'describe': '人'},
'bicycle': {'index': 1, 'id': 2, 'describe': '自行车'},
'car': {'index': 2, 'id': 3, 'describe': '车'},
'motorcycle': {'index': 3, 'id': 4, 'describe': '摩托车'},
'airplane': {'index': 4, 'id': 5, 'describe': '飞机'},
'bus': {'index': 5, 'id': 6, 'describe': '公交车'},
'train': {'index': 6, 'id': 7, 'describe': '火车'},
'truck': {'index': 7, 'id': 8, 'describe': '卡车'},
'boat': {'index': 8, 'id': 9, 'describe': '船'},
'traffic light': {'index': 9, 'id': 10, 'describe': '红绿灯'},
'fire hydrant': {'index': 10, 'id': 11, 'describe': '消防栓'},
'stop sign': {'index': 11, 'id': 13, 'describe': '停车标志'},
'parking meter': {'index': 12, 'id': 14, 'describe': '停车收费表'},
'bench': {'index': 13, 'id': 15, 'describe': '板凳'},
'bird': {'index': 14, 'id': 16, 'describe': '鸟'},
'cat': {'index': 15, 'id': 17, 'describe': '猫'},
'dog': {'index': 16, 'id': 18, 'describe': '狗'},
'horse': {'index': 17, 'id': 19, 'describe': '马'},
'sheep': {'index': 18, 'id': 20, 'describe': '羊'},
'cow': {'index': 19, 'id': 21, 'describe': '牛'},
'elephant': {'index': 20, 'id': 22, 'describe': '大象'},
'bear': {'index': 21, 'id': 23, 'describe': '熊'},
'zebra': {'index': 22, 'id': 24, 'describe': '斑马'},
'giraffe': {'index': 23, 'id': 25, 'describe': '长颈鹿'},
'backpack': {'index': 24, 'id': 27, 'describe': '背包'},
'umbrella': {'index': 25, 'id': 28, 'describe': '雨伞'},
'handbag': {'index': 26, 'id': 31, 'describe': '手提包'},
'tie': {'index': 27, 'id': 32, 'describe': '领带'},
'suitcase': {'index': 28, 'id': 33, 'describe': '手提箱'},
'frisbee': {'index': 29, 'id': 34, 'describe': '飞盘'},
'skis': {'index': 30, 'id': 35, 'describe': '雪橇'},
'snowboard': {'index': 31, 'id': 36, 'describe': '滑雪板'},
'sports ball': {'index': 32, 'id': 37, 'describe': '运动球'},
'kite': {'index': 33, 'id': 38, 'describe': '风筝'},
'baseball bat': {'index': 34, 'id': 39, 'describe': '棒球棒'},
'baseball glove': {'index': 35, 'id': 40, 'describe': '棒球手套'},
'skateboard': {'index': 36, 'id': 41, 'describe': '滑板'},
'surfboard': {'index': 37, 'id': 42, 'describe': '冲浪板'},
'tennis racket': {'index': 38, 'id': 43, 'describe': '网球拍'},
'bottle': {'index': 39, 'id': 44, 'describe': '瓶'},
'wine glass': {'index': 40, 'id': 46, 'describe': '酒杯'},
'cup': {'index': 41, 'id': 47, 'describe': '杯'},
'fork': {'index': 42, 'id': 48, 'describe': '叉子'},
'knife': {'index': 43, 'id': 49, 'describe': '刀'},
'spoon': {'index': 44, 'id': 50, 'describe': '汤匙'},
'bowl': {'index': 45, 'id': 51, 'describe': '碗'},
'banana': {'index': 46, 'id': 52, 'describe': '香蕉'},
'apple': {'index': 47, 'id': 53, 'describe': '苹果'},
'sandwich': {'index': 48, 'id': 54, 'describe': '三明治'},
'orange': {'index': 49, 'id': 55, 'describe': '橙'},
'broccoli': {'index': 50, 'id': 56, 'describe': '西兰花'},
'carrot': {'index': 51, 'id': 57, 'describe': '胡萝卜'},
'hot dog': {'index': 52, 'id': 58, 'describe': '热狗'},
'pizza': {'index': 53, 'id': 59, 'describe': '披萨'},
'donut': {'index': 54, 'id': 60, 'describe': '甜甜圈'},
'cake': {'index': 55, 'id': 61, 'describe': '蛋糕'},
'chair': {'index': 56, 'id': 62, 'describe': '椅子'},
'couch': {'index': 57, 'id': 63, 'describe': '沙发'},
'potted plant': {'index': 58, 'id': 64, 'describe': '盆栽植物'},
'bed': {'index': 59, 'id': 65, 'describe': '床'},
'dining table': {'index': 60, 'id': 67, 'describe': '餐桌'},
'toilet': {'index': 61, 'id': 70, 'describe': '厕所'},
'tv': {'index': 62, 'id': 72, 'describe': '电视'},
'laptop': {'index': 63, 'id': 73, 'describe': '笔记本电脑'},
'mouse': {'index': 64, 'id': 74, 'describe': '鼠标'},
'remote': {'index': 65, 'id': 75, 'describe': '遥控器'},
'keyboard': {'index': 66, 'id': 76, 'describe': '键盘'},
'cell phone': {'index': 67, 'id': 77, 'describe': '手机'},
'microwave': {'index': 68, 'id': 78, 'describe': '微波炉'},
'oven': {'index': 69, 'id': 79, 'describe': '烤箱'},
'toaster': {'index': 70, 'id': 80, 'describe': '烤面包机'},
'sink': {'index': 71, 'id': 81, 'describe': '水槽'},
'refrigerator': {'index': 72, 'id': 82, 'describe': '冰箱'},
'book': {'index': 73, 'id': 84, 'describe': '书'},
'clock': {'index': 74, 'id': 85, 'describe': '时钟'},
'vase': {'index': 75, 'id': 86, 'describe': '花瓶'},
'scissors': {'index': 76, 'id': 87, 'describe': '剪刀'},
'teddy bear': {'index': 77, 'id': 88, 'describe': '泰迪熊'},
'hair drier': {'index': 78, 'id': 89, 'describe': '吹风机'},
'toothbrush': {'index': 79, 'id': 90, 'describe': '牙刷'},
}