image_crawl/postprocess_folders.py

#!/usr/bin/env python
if __name__ != '__main__': raise Exception("Do not import me!")

import os
import numpy as np
import re
import hashlib
import logging

from cvargparse import Arg
from os.path import join, basename, dirname, isdir
from tqdm import tqdm

from collections import OrderedDict

from utils import parser
from utils import imaging

def rename(path, new_folder):
	parent = dirname(path)
	new_path = join(parent, new_folder)

	assert not isdir(new_path), f"Folder already exists: {new_path}!"
	# print(path, "->", new_path)
	os.rename(path, new_path)

	return new_path


def main(args):

	class_names = imaging.get_classnames(args.classnames)
	content = imaging.get_content(args.folder, args.extensions)

	for root, fnames in tqdm(content):
		folder = basename(root)

		if folder not in class_names:
			assert folder in class_names.values(), f"\"{folder}\" has not valid name!"

		fpaths = [join(root, fname) for fname in fnames]

		new_path = rename(root, class_names[folder])

		# import pdb; pdb.set_trace()

main(parser.parse_args([
	Arg("classnames")
]))