thesis_functions.py 文件源码

def plot_image_final_label(path, top_class, m):

    original_image = cv2.imread(path)

    # Swap Red and Blue color channels BGR -> RGB
    red = original_image[:, :, 2].copy()
    blue = original_image[:, :, 0].copy()
    original_image[:, :, 0] = red
    original_image[:, :, 2] = blue

    cv2.putText(original_image, "Label: {}".format(top_class), (10, 30),
                cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 0, 255), 2)
    plt.figure()
    plt.imshow(original_image)
    plt.savefig('final_solution_' + str(m) + '.jpg')

#########################################################################################

#########################################################################################
# FUNCTION: DRAW BAR GRAPH                                                              #
# GOAL:     Draw input image and graph bar with top 5 labels                            #
# INPUT:    final_top_5, final_top_5_prob, path, m                                      #
# OUTPUT:   Graph bar                                                                   #
# RETURN:                                                                               #
#########################################################################################