로컬 이미지로 데이터셋 만들기
#로컬 이미지로 데이터셋 만들기 import os, re, glob import cv2 #openCV 라이브러리 import하기 import numpy as np from sklearn.model_selection import train_test_split #현재 로컬 이미지 폴더 구조 #dataset/25/road, water, building, green imagePath = './dataset/25' categories = ["road", "water", "building", "green"] #dataset/25 하위 폴더의 이름이 카테고리가 됨. 동일하게 맞춰줘야한다. nb_classes = len(categories) image_w = 28 image_h = 28 X = [] Y = [] for..
AI
2019. 8. 26. 17:31