From abea3397efbf2d8088ae9b187ee0bbda11124f1c Mon Sep 17 00:00:00 2001 From: WZMIAOMIAO <605169423@qq.com> Date: Wed, 29 Dec 2021 18:18:35 +0800 Subject: [PATCH] fix plt.imshow --- ...\265\213\345\256\236\350\267\265\347\257\207.ipynb" | 2 +- ...\256\236\350\267\265\351\203\250\345\210\206.ipynb" | 8 ++++---- ...\206\346\236\220\345\256\236\346\210\230-VQA.ipynb" | 2 +- ...\241\250\346\240\274\350\257\206\345\210\253.ipynb" | 10 +++++----- 4 files changed, 11 insertions(+), 11 deletions(-) diff --git "a/notebook/notebook_ch/2.text_detection/\346\226\207\346\234\254\346\243\200\346\265\213\345\256\236\350\267\265\347\257\207.ipynb" "b/notebook/notebook_ch/2.text_detection/\346\226\207\346\234\254\346\243\200\346\265\213\345\256\236\350\267\265\347\257\207.ipynb" index 05060bef82..5a037dd79a 100644 --- "a/notebook/notebook_ch/2.text_detection/\346\226\207\346\234\254\346\243\200\346\265\213\345\256\236\350\267\265\347\257\207.ipynb" +++ "b/notebook/notebook_ch/2.text_detection/\346\226\207\346\234\254\346\243\200\346\265\213\345\256\236\350\267\265\347\257\207.ipynb" @@ -218,7 +218,7 @@ "\n", "# 画出读取的图片\n", "plt.figure(figsize=(10, 10))\n", - "plt.imshow(image)\n" + "plt.imshow(image[:, :, ::-1])\n" ] }, { diff --git "a/notebook/notebook_ch/3.text_recognition/\346\226\207\346\234\254\350\257\206\345\210\253\345\256\236\350\267\265\351\203\250\345\210\206.ipynb" "b/notebook/notebook_ch/3.text_recognition/\346\226\207\346\234\254\350\257\206\345\210\253\345\256\236\350\267\265\351\203\250\345\210\206.ipynb" index c8361cd456..b00b4517f6 100644 --- "a/notebook/notebook_ch/3.text_recognition/\346\226\207\346\234\254\350\257\206\345\210\253\345\256\236\350\267\265\351\203\250\345\210\206.ipynb" +++ "b/notebook/notebook_ch/3.text_recognition/\346\226\207\346\234\254\350\257\206\345\210\253\345\256\236\350\267\265\351\203\250\345\210\206.ipynb" @@ -445,12 +445,12 @@ "plt.figure()\n", "plt.subplot(2,1,1)\n", "# 可视化原图\n", - "plt.imshow(raw_img)\n", + "plt.imshow(raw_img[:, :, ::-1])\n", "# 缩放并归一化\n", "padding_im, draw_img = resize_norm_img(raw_img)\n", "plt.subplot(2,1,2)\n", "# 可视化网络输入图\n", - "plt.imshow(draw_img)\n", + "plt.imshow(draw_img[:, :, ::-1])\n", "plt.show()" ] }, @@ -1475,12 +1475,12 @@ "plt.figure()\n", "plt.subplot(2,1,1)\n", "# 可视化原图\n", - "plt.imshow(raw_img)\n", + "plt.imshow(raw_img[:, :, ::-1])\n", "# 随机切割\n", "crop_img = get_crop(raw_img)\n", "plt.subplot(2,1,2)\n", "# 可视化增广图\n", - "plt.imshow(crop_img)\n", + "plt.imshow(crop_img[:, :, ::-1])\n", "plt.show()" ] }, diff --git "a/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-VQA.ipynb" "b/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-VQA.ipynb" index 44f6f0f899..b5f500d07f 100644 --- "a/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-VQA.ipynb" +++ "b/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-VQA.ipynb" @@ -313,7 +313,7 @@ "\n", "img = cv2.imread('output/res_e2e/zh_val_42_ser.jpg')\n", "plt.figure(figsize=(48,24))\n", - "plt.imshow(img)" + "plt.imshow(img[:, :, ::-1])" ] }, { diff --git "a/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-\350\241\250\346\240\274\350\257\206\345\210\253.ipynb" "b/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-\350\241\250\346\240\274\350\257\206\345\210\253.ipynb" index b651636f20..62f82091f7 100644 --- "a/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-\350\241\250\346\240\274\350\257\206\345\210\253.ipynb" +++ "b/notebook/notebook_ch/6.document_analysis/\346\226\207\346\241\243\345\210\206\346\236\220\345\256\236\346\210\230-\350\241\250\346\240\274\350\257\206\345\210\253.ipynb" @@ -255,7 +255,7 @@ "\n", "# 读取表格图像并显示\n", "img = cv2.imread('/home/aistudio/1.jpg')\n", - "plt.imshow(img)" + "plt.imshow(img[:, :, ::-1])" ] }, { @@ -478,7 +478,7 @@ "# 显示输入图像\n", "plt.subplot(1,3,1)\n", "plt.title('src, shape:{}'.format(img.shape))\n", - "plt.imshow(img)\n", + "plt.imshow(img[:, :, ::-1])\n", "\n", "# 执行 ResizeTableImage\n", "# https://github.com/PaddlePaddle/PaddleOCR/blob/dygraph/ppocr/data/imaug/gen_table_mask.py#L182\n", @@ -491,7 +491,7 @@ "# 显示 ResizeTableImage 后的图像\n", "plt.subplot(1,3,2)\n", "plt.title('ResizeTableImage, shape:{}'.format(data['image'].shape))\n", - "plt.imshow(data['image'])\n", + "plt.imshow(data['image'][:, :, ::-1])\n", "\n", "# 执行 PaddingTableImage\n", "# https://github.com/PaddlePaddle/PaddleOCR/blob/dygraph/ppocr/data/imaug/gen_table_mask.py#L232\n", @@ -504,7 +504,7 @@ "# 显示 PaddingTableImage 后的图像\n", "plt.subplot(1,3,3)\n", "plt.title('PaddingTableImage, shape:{}'.format(data['image'].shape))\n", - "plt.imshow(data['image']/255)\n", + "plt.imshow(data['image'][:, :, ::-1]/255)\n", "plt.show()\n", "\n", "# 定义完整的处理op列表\n", @@ -902,7 +902,7 @@ "img_show = img.copy()\n", "for box in res_loc_final:\n", " cv2.rectangle(img_show, (box[0], box[1]), (box[2], box[3]), (0, 255, 0), 2)\n", - "plt.imshow(img_show)" + "plt.imshow(img_show[:, :, ::-1])" ] }, {