Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
HamedMP committed Jan 13, 2016
1 parent 15a913b commit 922581e
Showing 1 changed file with 27 additions and 1 deletion.
28 changes: 27 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,33 @@ pip install imageflow
```

Usage:
[Will be updated soon]

```python
import imageflow
```

#### Convert a directory of images and their labels to `.tfrecords`
Just calling the following function will make a `filename.tfrecords` file in the directory `converted_data` in your projects root(where you call this method).

```python
convert_images(images, labels, filename)
```

The `images` should be an array of shape `[-1, height, width, channel]` and has the same rows as the `labels`

#### Read distorted and normal dta from `.tfrecords` in multi-thread manner:
```python
# Distorted images for training
images, labels = distorted_inputs(filename='../my_data_raw/train.tfrecords', batch_size=FLAGS.batch_size,
num_epochs=FLAGS.num_epochs,
num_threads=5, imshape=[32, 32, 3], imsize=32)

# Normal images for validation
val_images, val_labels = inputs(filename='../my_data_raw/validation.tfrecords', batch_size=FLAGS.batch_size,
num_epochs=FLAGS.num_epochs,
num_threads=5, imshape=[32, 32, 3])
```


Dependencies:

Expand Down

0 comments on commit 922581e

Please sign in to comment.