s425850/AI_PRO
1
0
Fork 0
forked from s452751/AI_PRO
Code Issues Pull Requests Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
secret_dude 2021-06-02 01:32:48 +02:00
parent 4acea19558
commit c6df15c609
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
createNeuralNetworkDatabase.py
View File

@ -16,6 +16,7 @@ for img in os.listdir(path_potatoes):
pic = cv2.resize(pic,(size,size))
training_data_potatoes.append([pic])
np.save(os.path.join('neural_network','potatoes-dataset'),np.array(training_data_potatoes))
saved_potatoes = np.load(os.path.join('neural_network','potatoes-dataset.npy'))
@ -32,6 +33,10 @@ np.save(os.path.join('neural_network','beetroot-dataset'),np.array(training_data
saved_potatoes = np.load(os.path.join('neural_network','beetroot-dataset.npy'))
# for i in range(5):
# plt.imshow(np.array(training_data_potatoes[i]).reshape(size,size,3))
# plt.show()
dict = {
'beetroots': training_data_beetroot,
'potatoes': training_data_potatoes
}
print(dict)
np.save(os.path.join('neural_network','dataset'), np.array(dict))