obj bbox dataset slice

obj_slice.py

from PIL import Image
from pprint import pprint
import xmltodict

with open('pizza.xml') as fd:
    doc = xmltodict.parse(fd.read())

file_path = doc['annotation']['path']
main_img = Image.open(file_path)

obj_arr = doc['annotation']['object']
for idx, obj in enumerate(obj_arr):
    pprint(obj)
    name_class = obj['name']
    box = obj['bndbox']

    img = main_img.crop((int(box['xmin']), int(box['ymin']), int(box['xmax']), int(box['ymax']))) \
        .save('{}-{}.jpg'.format(name_class, idx))