斑马线预测-深度学习

这次我用的是卷积神经网络，来实现斑马线的预测

#导入库函数

import tensorflow as tf
import  matplotlib.pyplot as plt
import cv2 as cv
import pandas as pd
import numpy as np
import os
import shutil
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
from tensorflow.keras.models import Sequential, load_model
from tensorflow.keras import optimizers
from tensorflow.keras.preprocessing.image import ImageDataGenerator

train='E://Zebra/train'
val='E://Zebra/test'
train_data = ImageDataGenerator(rescale=1/255)
train_generator = train_data.flow_from_directory(train, (50, 50), batch_size=20, class_mode='binary', shuffle=True)
validation_data = ImageDataGenerator(rescale=1/255)
validation_generator = validation_data.flow_from_directory(val, (50, 50), batch_size=20, class_mode='binary')

#构建模型

if os.path.exists('pavement_plain.h5'):
    pri