import turicreate as tc
import os
import inspect

IMAGES_DIR = 'Images'
# data = 'test.json'

json_sf = tc.SFrame.read_json('test2.json')

sf_images = tc.image_analysis.load_images(IMAGES_DIR, recursive=True, random_order=False)

info = sf_images['path'].apply(lambda path: ['/'.join(path.split('/')[-2:])])

info = info.unpack().rename({'X.0': 'name'})

sf_images = sf_images.add_columns(info)

# json_sf.rename({'X.0': 'name'})

sf = sf_images.join(json_sf, on='name', how='left')

# json_sf.save('test.sframe')
sf['image_with_ground_truth'] = \
   tc.object_detector.util.draw_bounding_boxes(sf['image'], sf['annotations'])
sf.explore()
input('Pause...')