mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

decorate open by with

pull/1017/head
weishengyu 2021-07-07 14:30:45 +08:00
parent 0c3c266a3c
commit e4780174d1
4 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deploy/paddleserving/image_http_client.py
View File

@ -23,9 +23,11 @@ py_version = sys.version_info[0]
def predict(image_path, server): def predict(image_path, server):
if py_version == 2: if py_version == 2:
image = base64.b64encode(open(image_path).read()) with open(image_path) as f:
image = base64.b64encode(f.read())
else: else:
image = base64.b64encode(open(image_path, "rb").read()).decode("utf-8") with open(image_path, "rb") as f:
image = base64.b64encode(f.read()).decode("utf-8")
req = json.dumps({"feed": [{"image": image}], "fetch": ["prediction"]}) req = json.dumps({"feed": [{"image": image}], "fetch": ["prediction"]})
r = requests.post( r = requests.post(
server, data=req, headers={"Content-Type": "application/json"}) server, data=req, headers={"Content-Type": "application/json"})

10
docs/en/advanced_tutorials/image_augmentation/ImageAugment_en.md
View File

@ -154,10 +154,12 @@ cutout_op = Cutout(n_holes=1, length=112)
ops = [decode_op, resize_op, cutout_op] ops = [decode_op, resize_op, cutout_op]
imgs_dir = image_path imgs_dir = "imgs_dir"
fnames = os.listdir(imgs_dir) file_names = os.listdir(imgs_dir)
for f in fnames: for file_name in file_names:
data = open(os.path.join(imgs_dir, f)).read() file_path = os.join(imgs_dir, file_name)
with open(file_path) as f:
data = f.read()
img = transform(data, ops) img = transform(data, ops)
``` ```

10
docs/zh_CN/advanced_tutorials/image_augmentation/ImageAugment.md
View File

@ -74,10 +74,12 @@ autoaugment_op = ImageNetPolicy()
ops = [decode_op, resize_op, autoaugment_op] ops = [decode_op, resize_op, autoaugment_op]
imgs_dir = 图像路径 imgs_dir = "imgs_dir"
fnames = os.listdir(imgs_dir) file_names = os.listdir(imgs_dir)
for f in fnames: for file_name in file_names:
data = open(os.path.join(imgs_dir, f)).read() file_path = os.join(imgs_dir, file_name)
with open(file_path) as f:
data = f.read()
img = transform(data, ops) img = transform(data, ops)
``` ```

3
ppcls/data/dataloader/vehicle_dataset.py
View File

@ -61,7 +61,8 @@ class CompCars(Dataset):
label_path = os.path.join(self._label_root, label_path = os.path.join(self._label_root,
l[0].split('.')[0] + '.txt') l[0].split('.')[0] + '.txt')
assert os.path.exists(label_path) assert os.path.exists(label_path)
bbox = open(label_path).readlines()[-1].strip().split() with open(label_path) as f:
bbox = f.readlines()[-1].strip().split()
bbox = [int(x) for x in bbox] bbox = [int(x) for x in bbox]
self.images.append(os.path.join(self._img_root, l[0])) self.images.append(os.path.join(self._img_root, l[0]))
self.labels.append(int(l[1])) self.labels.append(int(l[1]))