From fa6c5517836db20a0e95af06a497870d70960c18 Mon Sep 17 00:00:00 2001 From: Adam Osiowy Date: Tue, 12 May 2020 12:21:31 +0200 Subject: [PATCH] aktualizacja tworzenia pliku z danymi h5 --- etykiety.h5 | Bin 17944 -> 17944 bytes parametry_zdjec.h5 | Bin 8458720 -> 8458720 bytes tworzenie_danych_AO.py | 36 +++++++++++++++++------------------- 3 files changed, 17 insertions(+), 19 deletions(-) diff --git a/etykiety.h5 b/etykiety.h5 index c157d34810a5c9c7f0101061d4e93393b6f4f0bc..c7604bc17e0343e0f17c4f035baac3d696738427 100644 GIT binary patch delta 25 hcmbQy!#Jadal;a3_LU{O;ushjCT?uqY{2ru5dedA3N-)# delta 25 hcmbQy!#Jadal;a3cC8yb;}{qkCT?uqY{2ru5deX23K;+Z diff --git a/parametry_zdjec.h5 b/parametry_zdjec.h5 index aa14a0d5eeee6ba86c09c96671e92ed3cdb4bb66..c2591ea016946836e26f0c82749f8d93c3050b1f 100644 GIT binary patch delta 536 zcmWm0Su>RZ007X@ckO%GLr4oMOUN2hLWS%~q)^$Hi-a(m@k2cI)|(k)=Kd8wrQ_ku zoU5}xp)3Af2EpI(#YrSMjRaBL;QAq=i6NFa;z=NpB$7!Xl{C`1$t^O-B#Ufv$R&^4 zhm=uH1&?^l6Dp~qni`%`OC9w*qk%@6Xr_f$p3_D<9lW5E zm%O5j*L2fEFK>9uJNoEnfcFeC#4sa_GR8O)eBdLKd}4}eJ~P7?zB0=k^L*nwKUiRq kC6-xXl{MDcV3ROgY_r2pcKO8~`yB9_LypcP!EtfQe`FfZjsO4v delta 536 zcmWm0*)o*@007|OM_D?llqKYpgp}-&H5AIeZy|fxDqEIlnCU^hfV;+wne!T5dXzpF z-^_P)b{ccV`DG9k{<}Dd1b-qy6wzEihFIdb!A;^xAdw_)5#lz<+~F?wNZ~#YNF|MQ zGI&TPS!DBw9Kt;23AsEak7wjlKq1d5;sq}$<`pHBQbsuyR8qxjs;S`(wbW5h1C2EC zmS$RLu@R3fs=%$BW`sinXL53J+gi*#AXM#ySF~u}9d}fw8=2>8oC6@WZ mS5{bMjdeEo#&>?O$rjt}u**;O*yn&lesRR{-$?K~9QqF`XwPy0 diff --git a/tworzenie_danych_AO.py b/tworzenie_danych_AO.py index 7031b98..8240f3d 100644 --- a/tworzenie_danych_AO.py +++ b/tworzenie_danych_AO.py @@ -30,41 +30,39 @@ def histogram(zdj, mask=None): return hist.flatten() -klasy = os.listdir(sciezka_do_zdj) +klasy = os.listdir(sciezka_do_zdj) # glass metal paper plastic klasy.sort() -dane = [] -wszystkie_typy_zdj = [] +dane = [] # parametry kazdego zdjecia + +etykiety_zdj = [] # typ kazdego zdjecia for klasa in klasy: katalog = os.path.join(sciezka_do_zdj, klasa) - biezaca_klasa = klasa for plik in os.listdir(katalog): - zdj = cv2.imread(os.path.join(katalog, plik)) - zdj = cv2.resize(zdj, rozmiar_zdj) + zdj = cv2.imread(os.path.join(katalog, plik)) # odczyt zdjecia + zdj = cv2.resize(zdj, rozmiar_zdj) # zmiana rozmiaru + # wyznaczanie parametrow zdjecia fv_hu_moments = hu_moments(zdj) fv_haralick = haralick(zdj) fv_histogram = histogram(zdj) - wiersz = np.hstack([fv_hu_moments, fv_histogram, fv_haralick]) - wszystkie_typy_zdj.append(biezaca_klasa) + wiersz = np.hstack([fv_hu_moments, fv_histogram, fv_haralick]) # ustaw poziomo, jeden za drugim + etykiety_zdj.append(klasa) dane.append(wiersz) -print(len(wszystkie_typy_zdj)) -targetNames = np.unique(wszystkie_typy_zdj) -print(targetNames,targetNames.shape) +# ilosc wszystkich etykiet to 1987 - tyle ile wszystkich zdj le = LabelEncoder() -target = le.fit_transform(wszystkie_typy_zdj) -print(target.shape) +znormalizowane_etykiety = le.fit_transform(etykiety_zdj) # normalizacja do 0-glass 1-metal itp. -h5f_data = h5py.File(h5_dane, 'w') -h5f_data.create_dataset('dataset_1', data=np.array(dane)) +h5f_dane = h5py.File(h5_dane, 'w') +h5f_dane.create_dataset('dataset_1', data=np.array(dane)) -h5f_label = h5py.File(h5_etykiety, 'w') -h5f_label.create_dataset('dataset_1', data=np.array(target)) +h5f_etykiety = h5py.File(h5_etykiety, 'w') +h5f_etykiety.create_dataset('dataset_1', data=np.array(znormalizowane_etykiety)) -h5f_data.close() -h5f_label.close() +h5f_dane.close() +h5f_etykiety.close()