resize_image.jl

addprocs()

src_path = "train"
dst_path = "/mnt/SSD/datasets/imagnet128"

mkpath(dst_path)

for folder in readdir(src_path)
    println("processing folder $folder")
    files = filter(f -> endswith(f, ".JPEG"), readdir(joinpath(src_path, folder)))
    args = [(joinpath(src_path, folder, f), joinpath(dst_path, folder, f)) for f in files]
    mkpath(joinpath(dst_path, folder))
    pmap(args) do fs
        try
            run(`convert $(fs[1]) -resize 128x128+0+0^ -gravity center -crop 128x128+0+0 $(fs[2])`)
        catch
            println(">> error in $(fs[1])")
        end
    end
end