Skip to content

Commit 990b7d7

Browse files
committed
rename xmap to xmap_readers and remove 'from cv2 import resize' in image.py
1 parent e62a4d7 commit 990b7d7

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

python/paddle/v2/dataset/flowers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,8 @@ def reader():
105105
for sample, label in itertools.izip(data, batch['label']):
106106
yield sample, int(label)
107107

108-
return paddle.reader.xmap(mapper, reader, cpu_count(), buffered_size)
108+
return paddle.reader.xmap_readers(mapper, reader,
109+
cpu_count(), buffered_size)
109110

110111

111112
def train(mapper=default_mapper, buffered_size=1024):

python/paddle/v2/image.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
import cv2
44
except ImportError:
55
cv2 = None
6-
7-
from cv2 import resize
86
import os
97
import tarfile
108
import cPickle
@@ -164,7 +162,7 @@ def resize_short(im, size):
164162
h_new = size * h / w
165163
else:
166164
w_new = size * w / h
167-
im = resize(im, (h_new, w_new), interpolation=cv2.INTER_CUBIC)
165+
im = cv2.resize(im, (h_new, w_new), interpolation=cv2.INTER_CUBIC)
168166
return im
169167

170168

python/paddle/v2/reader/decorator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
__all__ = [
1616
'map_readers', 'buffered', 'compose', 'chain', 'shuffle',
17-
'ComposeNotAligned', 'firstn', 'xmap'
17+
'ComposeNotAligned', 'firstn', 'xmap_readers'
1818
]
1919

2020
import itertools
@@ -230,7 +230,7 @@ class XmapEndSignal():
230230
pass
231231

232232

233-
def xmap(mapper, reader, process_num, buffer_size):
233+
def xmap_readers(mapper, reader, process_num, buffer_size):
234234
"""
235235
Use multiprocess to map samples from reader by a mapper defined by user.
236236
And this function contains a buffered decorator.

0 commit comments

Comments
 (0)