Skip to content

Commit 70c213a

Browse files
StephanieLarocquenotoraptor
authored andcommitted
specify that data augmentation is used
1 parent 6f867c6 commit 70c213a

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

doc/fcn_2D_segm.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,12 @@ There is a total of 912 images taken from 36 patients.
5555
* Test set : 8 patients and 182 frames
5656

5757
Each pixel is labelled between 2 classes : polype or background.
58-
The size of the images vary.
58+
The size of the images vary. We use data augmentation for training, as specified
59+
in the defaults arguments for data augmentation in the code given below. Note that
60+
the data augmentation is necessary for training with batch size greater than 1
61+
in order to have same image size with a random cropping. If no random cropping,
62+
the batch size for the training set must be set to 1, like for validation and test
63+
sets (where there is no data augmentation).
5964

6065

6166
In each of the training, validation and test directory, the input images are in the

doc/unet.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ Data
2020
++++
2121

2222
The data is from ISBI challenge and can be found `here <http://brainiac2.mit.edu/isbi_challenge/home>`_.
23+
We use data augmentation for training, as specified
24+
in the defaults arguments for data augmentation in the code given below.
2325

2426
Model
2527
+++++

0 commit comments

Comments
 (0)