测试用例实例

import os
import cv2
import sys
import glob
import numpy as np

def visualize_feature_map(img_batch,out_path,type,BI):
    feature_map = torch.squeeze(img_batch)
    feature_map = feature_map.detach().cpu().numpy()

    feature_map_sum = feature_map[0, :, :]
    feature_map_sum = np.expand_dims(feature_map_sum, axis=2)
    for i in range(0, 2048):
        feature_map_split = feature_map[i,:, :]
        feature_map_split = np.expand_dims(feature_map_split,axis=2)
        if i > 0:
            feature_map_sum +=feature_map_split
        feature_map_split = BI.transform(feature_map_split)

        plt.imshow(feature_map_split)
        plt.savefig(out_path + str(i) + "_{}.jpg".format(type) )
        plt.xticks()
        plt.yticks()
        plt.axis('off')

    feature_map_sum = BI.transform(feature_map_sum)
    plt.imshow(feature_map_sum)
    plt.savefig(out_path + "sum_{}.jpg".format(type))
    print("save sum_{}.jpg".format(type))

if __name__ == '__main__':
    imgs_path = "/path/to/imgs/"
    save_path = "/save/path/to/output/"
    model = Init_Setting(120)
    BI = BilinearInterpolation(8, 8)

    data = image_proprecess(out_path + "0836.jpg")
    data = data.cuda()
    output, _ = model(data)
    visualize_feature_map(output, save_path, "drone", BI)