1. 效果图
write pickle: coco_classes.pickle done
loading: coco_classes.pickle
['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush']
loading: coco.json
loading done.
coco json categories id [{
'supercategory': 'person', 'id': 1, 'name': 'person'}, {
'supercategory': 'vehicle', 'id': 2, 'name': 'bicycle'}, {
'supercategory': 'vehicle', 'id': 3, 'name': 'car'}, {
'supercategory': 'vehicle', 'id': 4, 'name': 'motorcycle'}, {
'supercategory': 'vehicle', 'id': 5, 'name': 'airplane'}, {
'supercategory': 'vehicle', 'id': 6, 'name': 'bus'}, {
'supercategory': 'vehicle', 'id': 7, 'name': 'train'}, {
'supercategory': 'vehicle', 'id': 8, 'name': 'truck'}, {
'supercategory': 'vehicle', 'id': 9, 'name': 'boat'}, {
'supercategory': 'outdoor', 'id': 10, 'name': 'traffic light'}, {
'supercategory': 'outdoor', 'id': 11, 'name': 'fire hydrant'}, {
'supercategory': 'outdoor', 'id': 13, 'name': 'stop sign'}, {
'supercategory': 'outdoor', 'id': 14, 'name': 'parking meter'}, {
'supercategory': 'outdoor', 'id': 15, 'name': 'bench'}, {
'supercategory': 'animal', 'id': 16, 'name': 'bird'}, {
'supercategory': 'animal', 'id': 17, 'name': 'cat'}, {
'supercategory': 'animal', 'id': 18, 'name': 'dog'}, {
'supercategory': 'animal', 'id': 19, 'name': 'horse'}, {
'supercategory': 'animal', 'id': 20, 'name': 'sheep'}, {
'supercategory': 'animal', 'id': 21, 'name': 'cow'}, {
'supercategory': 'animal', 'id': 22, 'name': 'elephant'}, {
'supercategory': 'animal', 'id': 23, 'name': 'bear'}, {
'supercategory': 'animal', 'id': 24, 'name': 'zebra'}, {
'supercategory': 'animal', 'id': 25, 'name': 'giraffe'}, {
'supercategory': 'accessory', 'id': 27, 'name': 'backpack'}, {
'supercategory': 'accessory', 'id': 28, 'name': 'umbrella'}, {
'supercategory': 'accessory', 'id': 31, 'name': 'handbag'}, {
'supercategory': 'accessory', 'id': 32, 'name': 'tie'}, {
'supercategory': 'accessory', 'id': 33, 'name': 'suitcase'}, {
'supercategory': 'sports', 'id': 34, 'name': 'frisbee'}, {
'supercategory': 'sports', 'id': 35, 'name': 'skis'}, {
'supercategory': 'sports', 'id': 36, 'name': 'snowboard'}, {
'supercategory': 'sports', 'id': 37, 'name': 'sports ball'}, {
'supercategory': 'sports', 'id': 38, 'name': 'kite'}, {
'supercategory': 'sports', 'id': 39, 'name': 'baseball bat'}, {
'supercategory': 'sports', 'id': 40, 'name': 'baseball glove'}, {
'supercategory': 'sports', 'id': 41, 'name': 'skateboard'}, {
'supercategory': 'sports', 'id': 42, 'name': 'surfboard'}, {
'supercategory': 'sports', 'id': 43, 'name': 'tennis racket'}, {
'supercategory': 'kitchen', 'id': 44, 'name': 'bottle'}, {
'supercategory': 'kitchen', 'id': 46, 'name': 'wine glass'}, {
'supercategory': 'kitchen', 'id': 47, 'name': 'cup'}, {
'supercategory': 'kitchen', 'id': 48, 'name': 'fork'}, {
'supercategory': 'kitchen', 'id': 49, 'name': 'knife'}, {
'supercategory': 'kitchen', 'id': 50, 'name': 'spoon'}, {
'supercategory': 'kitchen', 'id': 51, 'name': 'bowl'}, {
'supercategory': 'food', 'id': 52, 'name': 'banana'}, {
'supercategory': 'food', 'id': 53, 'name': 'apple'}, {
'supercategory': 'food', 'id': 54, 'name': 'sandwich'}, {
'supercategory': 'food', 'id': 55, 'name': 'orange'}, {
'supercategory': 'food', 'id': 56, 'name': 'broccoli'}, {
'supercategory': 'food', 'id': 57, 'name': 'carrot'}, {
'supercategory': 'food', 'id': 58, 'name': 'hot dog'}, {
'supercategory': 'food', 'id': 59, 'name': 'pizza'}, {
'supercategory': 'food', 'id': 60, 'name': 'donut'}, {
'supercategory': 'food', 'id': 61, 'name': 'cake'}, {
'supercategory': 'furniture', 'id': 62, 'name': 'chair'}, {
'supercategory': 'furniture', 'id': 63, 'name': 'couch'}, {
'supercategory': 'furniture', 'id': 64, 'name': 'potted plant'}, {
'supercategory': 'furniture', 'id': 65, 'name': 'bed'}, {
'supercategory': 'furniture', 'id': 67, 'name': 'dining table'}, {
'supercategory': 'furniture', 'id': 70, 'name': 'toilet'}, {
'supercategory': 'electronic', 'id': 72, 'name': 'tv'}, {
'supercategory': 'electronic', 'id': 73, 'name': 'laptop'}, {
'supercategory': 'electronic', 'id': 74, 'name': 'mouse'}, {
'supercategory': 'electronic', 'id': 75, 'name': 'remote'}, {
'supercategory': 'electronic', 'id': 76, 'name': 'keyboard'}, {
'supercategory': 'electronic', 'id': 77, 'name': 'cell phone'}, {
'supercategory': 'appliance', 'id': 78, 'name': 'microwave'}, {
'supercategory': 'appliance', 'id': 79, 'name': 'oven'}, {
'supercategory': 'appliance', 'id': 80, 'name': 'toaster'}, {
'supercategory': 'appliance', 'id': 81, 'name': 'sink'}, {
'supercategory': 'appliance', 'id': 82, 'name': 'refrigerator'}, {
'supercategory': 'indoor', 'id': 84, 'name': 'book'}, {
'supercategory': 'indoor', 'id': 85, 'name': 'clock'}, {
'supercategory': 'indoor', 'id': 86, 'name': 'vase'}, {
'supercategory': 'indoor', 'id': 87, 'name': 'scissors'}, {
'supercategory': 'indoor', 'id': 89, 'name': 'hair drier'}, {
'supercategory': 'indoor', 'id': 90, 'name': 'toothbrush'}, {
'supercategory': 'indoor', 'id': 88, 'name': 'teddy bear'}]
loading: coco_classes.pickle
names ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush']
2. 源码
read_write_pickle.py
import json
import pickle
# pickle 文件写入
def write_pickle(classes_pickle):
coco_list = ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase',
'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard',
'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl',
'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake',
'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote',
'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock',
'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush']
output = open(classes_pickle, 'wb')
pickle.dump(coco_list, output)
output.close()
print("write pickle:", str(classes_pickle), "done")
# load names
with open(classes_pickle, 'rb') as f:
print("loading:", str(classes_pickle))
names = pickle.load(f)
print(names)
# 读取 pickle文件
def coco_categories_names(categories, classes_pickle):
# load json names id
with open(categories) as f:
print('loading:', categories)
instances_val = json.load(f)
coco_categories = instances_val['categories']
print('coco json categories id', coco_categories)
# load names
with open(classes_pickle, 'rb') as f:
print("loading:", str(classes_pickle))
names = pickle.load(f)
print('names', names)
categories = 'coco.json'
# classes_pickle = 'coco_classes.pkl'
classes_pickle = 'coco_classes.pickle'
write_pickle(classes_pickle)
coco_categories_names(categories=categories, classes_pickle=classes_pickle)
coco.json
{
"categories":[{
"supercategory": "person", "id": 1, "name": "person"}, {
"supercategory": "vehicle", "id": 2, "name": "bicycle"}, {
"supercategory":
"vehicle", "id": 3, "name": "car"}, {
"supercategory": "vehicle", "id": 4, "name": "motorcycle"}, {
"supercategory": "vehicle", "id": 5, "name": "airplane"},
{
"supercategory": "vehicle", "id": 6, "name": "bus"}, {
"supercategory": "vehicle", "id": 7, "name": "train"}, {
"supercategory": "vehicle", "id": 8, "name": "truck"},
{
"supercategory": "vehicle", "id": 9, "name": "boat"}, {
"supercategory": "outdoor", "id": 10, "name": "traffic light"}, {
"supercategory": "outdoor", "id": 11, "name":
"fire hydrant"}, {
"supercategory": "outdoor", "id": 13, "name": "stop sign"}, {
"supercategory": "outdoor", "id": 14, "name": "parking meter"}, {
"supercategory":
"outdoor", "id": 15, "name": "bench"}, {
"supercategory": "animal", "id": 16, "name": "bird"}, {
"supercategory": "animal", "id": 17, "name": "cat"},
{
"supercategory": "animal", "id": 18, "name": "dog"}, {
"supercategory": "animal", "id": 19, "name": "horse"}, {
"supercategory": "animal", "id": 20, "name": "sheep"},
{
"supercategory": "animal", "id": 21, "name": "cow"}, {
"supercategory": "animal", "id": 22, "name": "elephant"}, {
"supercategory": "animal", "id": 23, "name":
"bear"}, {
"supercategory": "animal", "id": 24, "name": "zebra"}, {
"supercategory": "animal", "id": 25, "name": "giraffe"}, {
"supercategory": "accessory", "id": 27,
"name": "backpack"}, {
"supercategory": "accessory", "id": 28, "name": "umbrella"}, {
"supercategory": "accessory", "id": 31, "name": "handbag"}, {
"supercategory":
"accessory", "id": 32, "name": "tie"}, {
"supercategory": "accessory", "id": 33, "name": "suitcase"}, {
"supercategory": "sports", "id": 34, "name": "frisbee"},
{
"supercategory": "sports", "id": 35, "name": "skis"}, {
"supercategory": "sports", "id": 36, "name": "snowboard"}, {
"supercategory": "sports", "id": 37, "name":
"sports ball"}, {
"supercategory": "sports", "id": 38, "name": "kite"}, {
"supercategory": "sports", "id": 39, "name": "baseball bat"}, {
"supercategory":
"sports", "id": 40, "name": "baseball glove"}, {
"supercategory": "sports", "id": 41, "name": "skateboard"}, {
"supercategory": "sports", "id": 42, "name":
"surfboard"}, {
"supercategory": "sports", "id": 43, "name": "tennis racket"}, {
"supercategory": "kitchen", "id": 44, "name": "bottle"}, {
"supercategory":
"kitchen", "id": 46, "name": "wine glass"}, {
"supercategory": "kitchen", "id": 47, "name": "cup"}, {
"supercategory": "kitchen", "id": 48, "name": "fork"},
{
"supercategory": "kitchen", "id": 49, "name": "knife"}, {
"supercategory": "kitchen", "id": 50, "name": "spoon"}, {
"supercategory": "kitchen", "id": 51,
"name": "bowl"}, {
"supercategory": "food", "id": 52, "name": "banana"}, {
"supercategory": "food", "id": 53, "name": "apple"}, {
"supercategory": "food",
"id": 54, "name": "sandwich"}, {
"supercategory": "food", "id": 55, "name": "orange"}, {
"supercategory": "food", "id": 56, "name": "broccoli"},
{
"supercategory": "food", "id": 57, "name": "carrot"}, {
"supercategory": "food", "id": 58, "name": "hot dog"}, {
"supercategory": "food", "id": 59, "name": "pizza"},
{
"supercategory": "food", "id": 60, "name": "donut"}, {
"supercategory": "food", "id": 61, "name": "cake"}, {
"supercategory": "furniture", "id": 62, "name":
"chair"}, {
"supercategory": "furniture", "id": 63, "name": "couch"}, {
"supercategory": "furniture", "id": 64, "name": "potted plant"}, {
"supercategory":
"furniture", "id": 65, "name": "bed"}, {
"supercategory": "furniture", "id": 67, "name": "dining table"}, {
"supercategory": "furniture", "id": 70, "name":
"toilet"}, {
"supercategory": "electronic", "id": 72, "name": "tv"}, {
"supercategory": "electronic", "id": 73, "name": "laptop"}, {
"supercategory":
"electronic", "id": 74, "name": "mouse"}, {
"supercategory": "electronic", "id": 75, "name": "remote"}, {
"supercategory": "electronic", "id": 76, "name": "keyboard"},
{
"supercategory": "electronic", "id": 77, "name": "cell phone"}, {
"supercategory": "appliance", "id": 78, "name": "microwave"}, {
"supercategory": "appliance", "id":
79, "name": "oven"}, {
"supercategory": "appliance", "id": 80, "name": "toaster"}, {
"supercategory": "appliance", "id": 81, "name": "sink"}, {
"supercategory":
"appliance", "id": 82, "name": "refrigerator"}, {
"supercategory": "indoor", "id": 84, "name": "book"}, {
"supercategory": "indoor", "id": 85, "name": "clock"},
{
"supercategory": "indoor", "id": 86, "name": "vase"}, {
"supercategory": "indoor", "id": 87, "name": "scissors"}, {
"supercategory": "indoor", "id": 89, "name": "hair drier"}, {
"supercategory": "indoor", "id": 90, "name": "toothbrush"},{
"supercategory": "indoor", "id": 88, "name": "teddy bear"}]}