def plot_joints_from_colorspace(joints_colorspace, data=None, title='', fig=None, linewidth=2):
if fig is None:
fig = plt.figure()
fig = plot_joints(joints_colorspace, fig=fig, linewidth=linewidth)
if not data is None:
data_img_RGB = conv.numpy_to_plottable_rgb(data)
fig = plot_img_RGB(data_img_RGB, fig=fig, title=title)
return fig
def plot_image(data, title='', fig=None):
if fig is None:
fig = plt.figure()
data_img_RGB = conv.numpy_to_plottable_rgb(data)
plt.imshow(data_img_RGB)
if not title == '':
plt.title(title)
return fig
print_time('JORNet image conversion: ', time.time() - start)
start = time.time()
output_jornet = jornet(batch_jornet)
print_time('JORNet pass: '******'Plot image preparation: ', time.time() - start)
plt.imshow(conv.numpy_to_plottable_rgb(img_numpy))
plot_joints(labels_colorspace, show_legend=False)
total_elapsed_time = round(time.time() - start_beg, 2)
plt.title(input_img_namebase + ' : ' + str(total_elapsed_time) + ' ms')
plt.pause(0.001)
plt.clf()
print(
'--------------------------------------------------------------------------'
)
plt.show()