Сценарий обнаружения края Conv2D CNN возвращает пустые изображения - PullRequest
Новая
       35

Сценарий обнаружения края Conv2D CNN возвращает пустые изображения

0 голосов
/ 20 мая 2019

Я хочу покрасить масло, воду и пластик на показанных изображениях.

В настоящее время я разделил тренировочный образ (только используя части, которые окрашены правильно). Затем я обучаю сеть conv2D и строю прогнозы.

Когда я запускаю его, я получаю взамен пустые синие или черные изображения.

Посоветуйте пожалуйста:

  1. - подходит ли моя сеть,
  2. - какие параметры мне следует использовать
  3. - какие тренировочные образы мне следует использовать.

Input image Partially complete image used for training

#IMPORT AND SPLIT

from cam_img_split import cam_img_split
import cv2
import numpy as np

img_tr_in=cv2.imread('frame 1.png')
img_tr_out=cv2.imread('Red edge.png')

seg_shape=[32,32]

tr_in=cam_img_split(img_tr_in,seg_shape)
tr_out=cam_img_split(img_tr_out,seg_shape)

pl=[4,20]

##################### NEURAL NETWORK

import tensorflow as tf
from keras.models import Sequential
from keras.layers import Dense,Dropout,Conv2D, MaxPooling2D
from keras.optimizers import adam
import matplotlib.pyplot as plt

pad=3

input_shape=(seg_shape[0]+2*pad,seg_shape[1]+2*pad,3)

model = Sequential()
model.add(Conv2D(32, (3, 3),input_shape=input_shape, activation='relu'))
model.add(Conv2D(64,(3, 3), activation='relu')) 
model.add(Conv2D(3, (3, 3),input_shape=input_shape, activation='relu'))

model.compile(optimizer=adam(lr=0.001), loss='mean_squared_error', metrics=['accuracy'])

tr_in_sel=tr_in[0:pl[0],0:pl[1],:,:,:]
tr_out_sel=tr_out[0:pl[0],0:pl[1],:,:,:]

tr_in_sel_flat=tr_in_sel.reshape([pl[0]*pl[1],seg_shape[0],seg_shape[1],3])
tr_out_sel_flat=tr_out_sel.reshape([pl[0]*pl[1],seg_shape[0],seg_shape[1],3])

tr_in_sel_flat_norm=tr_in_sel_flat/255
tr_out_sel_flat_norm=tr_out_sel_flat/255

from cam_pad import cam_pad

tr_in_sel_flat_norm_pad=np.zeros(tr_in_sel_flat.shape+np.array([0,2*pad,2*pad,0]))

for n3 in range(0,tr_in_sel_flat.shape[0]):
    for n4 in range(0,tr_in_sel_flat.shape[3]):
        tr_in_sel_flat_norm_pad[n3,:,:,n4]=cam_pad(tr_in_sel_flat_norm[n3,:,:,n4], pad)

model.fit(tr_in_sel_flat_norm_pad, tr_out_sel_flat_norm, epochs=10, batch_size=int(pl[0]/2),shuffle=True)

n_ch=10
img_check=np.zeros([n_ch,seg_shape[0]+2*pad,seg_shape[1]+2*pad,3])

for n8 in range(0,n_ch):
    for n5 in range(0,3):
        img_check[n8,:,:,n5]=cam_pad(tr_in_sel_flat_norm[n8,:,:,n5],pad)

pred = model.predict(img_check/255)
pred_img=(pred.reshape([n_ch,seg_shape[0],seg_shape[1],3]))

for n9 in range(1,n_ch):
    plt.subplot(n_ch,1,n9)
    plt.imshow(pred_img[n9-1,:,:,:])

plt.show()

1 Ответ

1 голос
/ 22 мая 2019

Я случайно нормализовал (разделил на 255) мои данные дважды. Это приводит к обучению сети против пустых изображений и, следовательно, созданию пустых изображений.

...