This commit is contained in:
Kamila Bobkowska 2020-05-24 14:18:37 +00:00
parent 2eb83a3d02
commit 9e608db67c

View File

@ -57,11 +57,6 @@ trainingFileNames, testFileNames = np.split(np.array(allFileNames), [int(len(all
## Implementation
Functions in garbageDumpSorting.py:
* createSets - divides images between test and training set. This function should be run only once, unless the folders with training and test set are removed,
```
trainingFileNames, testFileNames = np.split(np.array(allFileNames), [int(len(allFileNames) * (1 - testRatio))])
```
* huMoments, haralick, histogram - calculate global feature descriptors,
* processTrainData, processTestData - both work in the same way, they iterate over files in train or test directory, saves features as a matrix and then saves results to h5 file, it is recommended to run it only once as it takes some time to finish.