-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
32 lines (22 loc) · 791 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import os
import shutil
import csv
import time
import math
import sys
from PIL import Image
import cv2
from os.path import abspath
from utils import build_scaled, crop, rgb, dbc
curr = os.path.dirname(__file__)
images_in = os.path.join(curr, r'data/image-data/imagini')
images_out = os.path.join(curr, r'data/image-data/images/resized')
build_scaled(images_in, images_out)
# ---------------------------------------------------------------------
images_in = images_out
images_out = os.path.join(curr, r'data/image-data/images/cropped')
crop(images_in, images_out)
# ---------------------------------------------------------------------
images_in = images_out
images_out = os.path.join(curr, r'data/image-data/images/rgb/')
rgb(images_in, images_out)