summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorEvan Shelhamer <shelhamer@imaginarynumber.net>2014-01-27 15:17:45 -0800
committerEvan Shelhamer <shelhamer@imaginarynumber.net>2014-01-27 15:17:45 -0800
commit47f8f536069be97da7bcd411070a71c6fad25fdb (patch)
treebe340f9e78621c1b8535c90dfb60ffbdc5a724d9 /python
parentb31b3167d5d28671f7a90aee47c969c9b51d54f1 (diff)
downloadcaffeonacl-47f8f536069be97da7bcd411070a71c6fad25fdb.tar.gz
caffeonacl-47f8f536069be97da7bcd411070a71c6fad25fdb.tar.bz2
caffeonacl-47f8f536069be97da7bcd411070a71c6fad25fdb.zip
replace magic numbers with variable names in comments
Diffstat (limited to 'python')
-rw-r--r--python/caffe/detection/detector.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/python/caffe/detection/detector.py b/python/caffe/detection/detector.py
index 26515389..034de3e9 100644
--- a/python/caffe/detection/detector.py
+++ b/python/caffe/detection/detector.py
@@ -48,7 +48,7 @@ def load_image(filename):
filename: string
Output:
- image: an image of size (256 x 256 x 3) of type uint8.
+ image: an image of size (IMAGE_DIM x IMAGE_DIM x 3) of type uint8.
"""
img = skimage.io.imread(filename)
if img.ndim == 2:
@@ -199,7 +199,7 @@ def assemble_batches(image_fnames, crop_mode='center_only'):
Output:
df_batches: list of DataFrames, each one of BATCH_SIZE rows.
Each row has 'image', 'filename', and 'window' info.
- Column 'image' contains (X x 3 x 227 x 227) ndarrays.
+ Column 'image' contains (X x 3 x CROPPED_DIM x CROPPED_IM) ndarrays.
Column 'filename' contains source filenames.
If 'filename' is None, then the row is just for padding.