if args.square:
if max_size == default_size:
max_size = default_square_size
else:
if max_size > 100:
print("Square mode, and size is {0}".format(max_size))
print("Error : Square size is too big.")
sys.exit(1)
head, tail = os.path.split(input_path)
dir_path = os.path.join(head, tail + "_dataset")
ds = DSUtils()
ds.files_copy(input_path, dir_path)
ds.files_change_ext(dir_path, ".jpeg", ".jpg")
ds.img_convert_ext(dir_path, ".jpg")
ds.files_lower_ext(dir_path)
ds.files_remove_notjpg(dir_path)
if not args.keepsize and args.square:
ds.img_square(dir_path, max_size)
elif not args.keepsize:
ds.img_resize(dir_path, max_size)
if not args.nodup:
rdd.images_remove_duplicated(dir_path, "")
ds.files_numbering(dir_path, name_header)
import os
import sys
# add module path
script_path = os.path.dirname(os.path.realpath(__file__))
module_path = os.path.join(script_path, "myutils")
sys.path.append(module_path)
# mymodules
import remove_dataset_duplicated as rdd
# main function
if __name__ == '__main__':
argv = sys.argv
argc = len(argv)
if argc < 2:
print("Error: The number of arguments must be one.")
print("arg 1: The source directory contains images.")
print("arg 2: The directory for comparison images. (option)")
sys.exit(1)
dir_path = os.path.normpath(argv[1])
cmp_path = []
if argc > 2:
cmp_path = [os.path.normpath(x) for x in argv[2:]]
rdd.images_remove_duplicated(dir_path, cmp_path)