【Python脚本】绘制常见激活函数代码\显示试验可视化对比进行拼接

使用python工具进行图片快速可视化对比

• 加载实验中可视化的图片后，对不同方法进行对比，对多个图像进行横向拼接，保存图片进行对比。
  代码如下

import os
import cv2
from tqdm import tqdm
import numpy as np

array_of_img = [] 

def read_directory(directory_name, data_dir):
    for filename in tqdm(os.listdir(data_dir)):
        img1 = cv2.imread(directory_name + "/xvo_kitti_test/vis/" + filename)
        img2 = cv2.imread(directory_name + "/sdp_kitti_test/vis/" + filename)
        img3 = cv2.imread(data_dir + "/" + filename)

        img1 = cv2.resize(img1,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)
        img2 = cv2.resize(img2,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)
        img3 = cv2.resize(img3,dsize=(416,128),fx=1,fy=1,interpolation=cv2.INTER_LINEAR)

        heng = np.hstack((img3, img1, img2))
        cv2.imwrite(directory_name + "/compara_result/" + filename, heng)

def main():
    img_dir = "/root/program/demo/output"
    data_dir = "/root/program/data/testing/color"
    read_directory(img_dir, data_dir)

if __name__ == '__main__':
    main()

绘制常见激活函数代码

import matplotlib.pyplot as plt
import numpy as np
import math

def elu(x,alpha=1):
    a = x[x>0]
    b = alpha*(math.e**(x[x<0])-1)
    result=np.concatenate((b,a),axis=0)
    return result

def relu(x):
    if x >= 0:
        return x
    else :
        return np.exp(x)-1
    #return np.maximum(0,x)

x = np.arange(-10, 10, 0.1)
y=elu(x)
plt.title("ELU")
plt.plot(x, y)
plt.show()