cat all filters in target path.

cat_filters.py

import os
import sys
import glob
import copy

from PIL import Image, ImageDraw, ImageFont

import torch
import torchvision
torch.backends.cudnn.benchmark=True
from torch import nn, optim
from torch.autograd import grad, Variable
from torch.autograd.gradcheck import zero_gradients
from torchvision import models
from torchvision.utils import save_image
from torchvision import transforms

base = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../../../')
sys.path.append(base)

#from misc import *

def cat_filters(path, nrow):
	paths = sorted(glob.glob(os.path.join(path, '*.png')))
	img_list = []

	transform = transforms.Compose([
		transforms.Resize(128),
		transforms.ToTensor(),	
	])

	for path in paths:
		img = Image.open(path).convert('RGB')
		img = transform(img)
		img_list.append(img)

	img_out = torch.stack(img_list, dim=0)
	img_out = torchvision.utils.make_grid(img_out, nrow, padding=1)

	return img_out

def main(trg_path):
	paths = glob.glob(trg_path)

	nrow = 10

	for path in paths:
		name = path.split('/')[-1]

		if name in ['archive'] or name.split('.')[-1] in ['py', 'png']:
			continue

		print(path)
		save_path = os.path.join(path, '..', name+'.png')
		
		if os.path.exists(save_path) and False:
			print('continue')
			continue

		img_out = transforms.ToPILImage()(cat_filters(path, nrow))
		font_name = '/home/gatheluck/Downloads/SansSerifBookFLF.otf'
		font = ImageFont.truetype(font_name, size=150)
		draw = ImageDraw.Draw(img_out)
		draw.text((30,30), name, font=font)
		img_out.save(save_path)
		
if __name__ == "__main__":
	trg_path = '/home/gatheluck/Scratch/arf/logs/uar/activation/*'
	main(trg_path)