 Stephen
|
c829284979
Added the validation set back in. I forgot that model.fit() can only automatically split a dataset into train/validation automatically if the input is not of the tf.whatever.dataset type. Which is dumb.
|
před 2 roky |
|
Stephen
|
f84119cbc7
Minor edits. Updated README.
|
před 2 roky |
|
Stephen
|
e2d9b9e575
Made creating folders simpler. Made folder for saving augmented dataset as pngs.
|
před 2 roky |
|
Stephen
|
b3998e930b
Created a funcion to save pictures to a file.
|
před 2 roky |
|
Stephen
|
1e698ddd99
Changed the name of validation set to testing set.
|
před 2 roky |
|
Stephen
|
0f6d096238
Removed training step from this project.
|
před 2 roky |
|
Stephen
|
5452034cbf
Removed creation of test set.
|
před 2 roky |
|
Stephen
|
04f3c7aad6
Reverted to grayscale.
|
před 2 roky |
|
Stephen
|
f8e9180fb5
Janky working version. Increase epochs and omit steps per epoch to get better results.
|
před 3 roky |
|
Stephen
|
3042281cf3
Working version with good accuracy in training. Need to make test set evaluation now.
|
před 3 roky |
|
Stephen
|
94b1e0155b
CHanged imagesize to a hardcoded 150 for the time being. Working version of data import on train script.
|
před 3 roky |
|
Stephen
|
03451ac13c
Changed input to rgb because most image models use rgb as input, can convert to grayscale before use if needed. Downside is the filesize is three times bigger. Added a script for training.
|
před 3 roky |
|
Stephen
|
a42423fa4b
Reduced image size back to 100 for now.
|
před 3 roky |
|
Stephen
|
7600aedc13
Added printing of the augmented data so we can see it. Added minor upgrades to parallelization.
|
před 3 roky |
|
Stephen
|
4b980bfcc5
Found how to get the images out of the prefetchdataset or whatever. Unclean running version.
|
před 3 roky |
|
Stephen
|
d9bfcfcf6c
Updated readme; The loader function shuffles automatically.
|
před 3 roky |
|
Stephen
|
1c08195815
Added tqdm for nice loading bar animations. Added a helper function for printing. I don't know how to print from the final datasets; they are of a different type.
|
před 3 roky |
|
Stephen
|
2818995099
CHanged picture input to grayscale. Fixed printing error.
|
před 3 roky |
|
Stephen
|
a90148dd4d
Remove old dataset files before running the script-- so tha they don't accumulate and waste space.
|
před 3 roky |
|
Stephen
|
a46b62c287
Enabled saving the dataset.
|
před 3 roky |
|
Stephen
|
d01407d623
Updated directory name
|
před 3 roky |
|
Stephen
|
1e132676f8
Removed un-needed loop in the image copy phase. Fixed a bug that would download the pictures multiple times if muliple triggers were set.
|
před 3 roky |
|
Stephen
|
7155b7230f
Fixed preprocessing. I found out that my tensorflow was really old-- causing some of the code to not work. Using tensorflow 2.7 and numpy 1.22.3 fixed it.
|
před 3 roky |
|
Stephen
|
319e2f6b02
Added a thing to automatically download if it is the first run. Also download flags from the command line and as a global variable.
|
před 3 roky |
|
Stephen
|
308d5f521f
Merge branch 'main' into mychanges
|
před 3 roky |
|
Stephen
|
9ab974765a
Added some STUFF.
|
před 3 roky |
 AlexWorland
|
a1148553b8
Added function that retrieves images if user hasnt yet
|
před 3 roky |
|
Stephen
|
14cdf0c54b
Added saving the datasets. It's quite large on disc though. Also added splitting off a test set.
|
před 3 roky |
|
Stephen
|
4d48f0ac4d
Load data and normalization complete. still need kfolds and test set.
|
před 3 roky |
|
Stephen
|
a36cc6e1d5
Success in copying staset into nice folder structure for yes/no machinelearning problem.
|
před 3 roky |
