Skip to content

Instantly share code, notes, and snippets.

@timehaven
Created July 19, 2017 15:39
Show Gist options
  • Save timehaven/04bc81e5d60a32c6134fcb843ef20443 to your computer and use it in GitHub Desktop.
Save timehaven/04bc81e5d60a32c6134fcb843ef20443 to your computer and use it in GitHub Desktop.
python classifier_from_little_data_script_dfgen.py
Using TensorFlow backend.
9936 train dog
9936 train cat
2564 validation dog
2564 validation cat
Some samples:
df_train:
object_id imgpath target orig label
7034 9898 /tmp/path/to/imgs/898/09/dog_9898.jpg 1 data/train/dogs/dog.9898.jpg dog
9126 3843 /tmp/path/to/imgs/843/03/dog_3843.jpg 1 data/train/dogs/dog.3843.jpg dog
12478 7431 /tmp/path/to/imgs/431/07/cat_7431.jpg 0 data/train/cats/cat.7431.jpg cat
43 9758 /tmp/path/to/imgs/758/09/dog_9758.jpg 1 data/train/dogs/dog.9758.jpg dog
19081 6499 /tmp/path/to/imgs/499/06/cat_6499.jpg 0 data/train/cats/cat.6499.jpg cat
df_valid:
object_id imgpath target orig label
4893 6152 /tmp/path/to/imgs/152/06/cat_6152.jpg 0 data/validation/cats/cat.6152.jpg cat
4496 143 /tmp/path/to/imgs/143/00/cat_143.jpg 0 data/validation/cats/cat.143.jpg cat
3362 1413 /tmp/path/to/imgs/413/01/cat_1413.jpg 0 data/validation/cats/cat.1413.jpg cat
1853 12307 /tmp/path/to/imgs/307/12/dog_12307.jpg 1 data/validation/dogs/dog.12307.jpg dog
2483 8613 /tmp/path/to/imgs/613/08/dog_8613.jpg 1 data/validation/dogs/dog.8613.jpg dog
Training set: 19872 images, 2 classes.
Validation set: 5128 images, 2 classes.
Epoch 1/5
name: Tesla K80
major: 3 minor: 7 memoryClockRate (GHz) 0.8235
pciBusID 0000:00:1e.0
Total memory: 11.17GiB
Free memory: 11.11GiB
88s - loss: 0.5950 - acc: 0.6865 - val_loss: 0.4879 - val_acc: 0.7721
Epoch 2/5
74s - loss: 0.5032 - acc: 0.7698 - val_loss: 0.4824 - val_acc: 0.7779
Epoch 3/5
74s - loss: 0.4663 - acc: 0.7911 - val_loss: 0.4875 - val_acc: 0.7826
Epoch 4/5
74s - loss: 0.4403 - acc: 0.8103 - val_loss: 0.4486 - val_acc: 0.8057
Epoch 5/5
74s - loss: 0.4209 - acc: 0.8198 - val_loss: 0.3943 - val_acc: 0.8273
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment