You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: ReadMe.md
+14-13Lines changed: 14 additions & 13 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,23 +1,24 @@
1
1
# Tensorflow CNN's
2
-
This library contains base methods for training various models in tensorflow using the following interface:
2
+
This library contains base methods for training various models in [TensorFlow](https://github.com/tensorflow/tensorflow) using the following interface:
3
3
4
-
```
5
-
import tfmodels
6
-
dataset = tfmodels.ImageMaskDataSet(...)
7
-
model = tfmodels.VGGTraining(datset, ...)
8
-
for _ in ...:
9
-
model.train()
4
+
```
5
+
import tfmodels
6
+
dataset = tfmodels.ImageMaskDataSet(...)
7
+
model = tfmodels.VGGTraining(datset, ...)
8
+
for _ in xrange(iterations):
9
+
model.train()
10
10
11
-
model.snapshot()
11
+
model.test()
12
+
model.snapshot()
12
13
13
-
result = model.inference(images)
14
-
```
14
+
result = model.inference(images)
15
+
```
15
16
16
-
To implement a new model, copy-paste one of the existing models and re-implement the `model()` method.
17
+
To implement a new `segmentation`model, copy-paste `segmentation/TEMPLATE.py` and fill in the `__init__`and `model` methods.
17
18
18
19
In addition to semantic segmentation models, the library also contains base methods for training generative models including Generative Adversarial Networks and Variational Autoencoders, and for multiple-instance classifiers.
19
20
20
-
**Note** the default activation (set in `tfmodels/utilities/basemodel.py`) is SeLU. Accordingly, the inputs should be scaled to `[-1.0, 1.0]` in the dataset loading functions, and we should use `tf.contrib.nn.alpha_dropout` (TensorFlow 1.4.1).
21
+
**Note** the default activation (set in `tfmodels/utilities/basemodel.py`) is [SeLU](https://arxiv.org/abs/1706.02515). Accordingly, the inputs should be scaled to `[-1.0, 1.0]` in the dataset loading functions, and we should use `tf.contrib.nn.alpha_dropout` (TensorFlow 1.4.1).
21
22
22
23
## Versioning
23
24
```
@@ -29,7 +30,7 @@ opencv 3
29
30
30
31
## Getting started
31
32
Example scripts for data set interface, training and testing various models are provided under `experiments/`.
32
-
- N-class semantic segmentation
33
+
- N-class semantic segmentation with various architectures (your data)
0 commit comments