diff --git a/img2bin/Readme.md b/img2bin/Readme.md index f255ac4..51644b3 100644 --- a/img2bin/Readme.md +++ b/img2bin/Readme.md @@ -3,43 +3,11 @@ img2bin能够生成模型推理所需的输入数据，以.bin格式保存。 有两类数据，一类是图片数据，另一类是模型需要的第二个输入数据，如fasterrcnn的第二个输入是图片的shape信息。 -## 环境准备 +## 前提条件 -执行此脚本需要安装python3的opencv，**如已安装可跳过环境准备**。 +脚本可在Centos和Ubuntu环境下使用，需提前安装python3和pip3。 -1. 在root用户下更换源。 - ``` - vim /etc/apt/sources.list - ``` - 把原有的源更换为国内可用的源，arm源可参考 https://bbs.huaweicloud.com/forum/thread-61366-1-1.html 。 - - - 源更新后，执行以下命令更新软件列表。 - - ``` - apt-get update - ``` - - - -2. 安装python3的依赖。 - - ``` - apt-get install python3-setuptools python3-dev build-essential python3-pip - ``` - ``` - pip3 install enum34==1.1.6 future==0.17.1 funcsigs==1.0.2 unique protobuf numpy - ``` - - >**说明：** - > - > pip3 install安装有报错“SSLError”时，请使用：pip3 install --trusted-host pypi.org --trusted-host files.pythonhosted.org numpy==1.11.2 enum34==1.1.6 future==0.17.1 funcsigs==1.0.2 unique protobuf 安装依赖，表示可信赖的主机解决问题。 - -3. 安装opencv。 - - ``` - pip3 install opencv_python - ``` +如未安装opencv-python，脚本会自动安装。 ## 获取脚本 @@ -58,11 +26,10 @@ img2bin能够生成模型推理所需的输入数据，以.bin格式保存。 ## 使用方法 进入脚本所在目录。 ``` -cd $HOME/AscendProjects/tools/img2bin +cd $HOME/AscendProjects/img2bin ``` ### 第一类图片： -- -i后跟**目录**表示转换图片。 - 脚本会将 -i 后指定的图片目录下的所有图片按参数设置做相应的预处理，并以"文件名.bin"命名保存在-o指定的输出目录下。 ``` @@ -70,10 +37,10 @@ python3 img2bin.py -i ./images -w 416 -h 416 -f BGR -a NHWC -t uint8 -m [104,117 ``` ### 第二类： -- -i后跟**文件路径**表示转换第二类数据。 -- 第二类数据，需要新建一个文件，文件模板为test.txt，"input_node"为数据，"shape"为数据的shape信息。 +- 第二类数据，需要新建一个文件，文件模板为test.txt，"input_node"为数据，"shape"为数据的shape信息。 +- 文件模板的后缀必须是“.txt”。 - 第二类数据只需 -i、-t、-o三个参数。 -- 参数 -i 需要指定文件的路径，-t 需要指定数据类型，-o指定输出目录。 +- 参数 -i 需要指定文件的目录或路径，-t 需要指定数据类型，-o指定输出目录。 ``` python3 img2bin.py -i ./test.txt -t uint8 -o ./out @@ -83,7 +50,7 @@ python3 img2bin.py -i ./test.txt -t uint8 -o ./out | 参数名 | 说明 | | - | - | -| -i | 图片的输入目录或第二个输入的文件路径 | +| -i | 输入目录或路径，
**在目录下，不能同时有图片和txt，一次只能转一种数据** | | -w | 输出图片宽 | | -h | 输出图片高 | | -f | 输出图片色彩格式，支持（BGR/RGB/YUV/GRAY） | diff --git a/img2bin/img2bin.py b/img2bin/img2bin.py index 40ff247..2eadd80 100644 --- a/img2bin/img2bin.py +++ b/img2bin/img2bin.py @@ -31,23 +31,28 @@ # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. # ======================================================================= -# import argparse import configparser -import cv2 as cv -import numpy as np import json import os -import re import sys - +try: + import cv2 as cv +except: + os.system('pip3 install --trusted-host pypi.org --trusted-host files.pythonhosted.org opencv-python') + import cv2 as cv +try: + import numpy as np +except: + os.system('pip3 install --trusted-host pypi.org --trusted-host files.pythonhosted.org numpy') + import numpy as np def get_args(): parser = argparse.ArgumentParser( conflict_handler='resolve', description='''eg1: python3 imgtobin.py - -i ./images -w 416 -h 416 -f BGR -a NCHW -m [104,117,123] -o ./out - eg2: python3 imgtobin.py -i ./test.txt -t uint8''') + -i ./images -w 416 -h 416 -f BGR -a NHWC -t uint8 -m [104,117,123] -c [1,1,1] -o ./out + eg2: python3 imgtobin.py -i ./test.txt -t uint8 -o ./out''') parser.add_argument('-i', '--input', required=True, type=str, \ help='folder of input image or file of other input.') parser.add_argument('-w', '--width', type=int, \ @@ -82,30 +87,19 @@ def check_args(args): check_flag = True is_dir = True if os.path.isdir(args.input): - #print(args) if not os.listdir(args.input): - eprint('[ERROR] input image path=%r is empty.' % path) + eprint('[ERROR] input image path=%r is empty.' % args.input) check_flag = False elif os.path.isfile(args.input): is_dir = False else: - eprint('[ERROR] input path=%r does not exist.' % path) + eprint('[ERROR] input path=%r does not exist.' % args.input) check_flag = False - if args.output_image_format not in ('BGR','RGB', 'YUV', 'GRAY'): eprint("ERROR:Convert to %d is not support"%(args.output_image_format)) check_flag = False - # if os.path.isfile(args.output_path): - # eprint('[ERROR] argument output_path should be a folder.') - # elif not os.path.exists(args.output_path): - # os.makedirs(args.output_path) - # if not 16 <= args.model_width <= 4096: - # eprint('[ERROR] resized image width should between 16 and 4096.') - # check_flag = False - # if not 16 <= args.model_height <= 4096: - # eprint('[ERROR] resized image height should between 16 andd 4096.') - # check_flag = False return check_flag, is_dir + def convert_img(args, input_img): @@ -116,11 +110,11 @@ def convert_img(args, input_img): elif args.output_image_format == 'YUV': if input_img.shape[0] % 2 == 1: if input_img.shape[1] % 2 == 1: - input_img = cv.resize(input_img, ((input_img.shape[0] + 1), (input_img.shape[1] + 1))) + input_img = cv.resize(input_img, ((input_img.shape[0] + 1), (input_img.shape[1] + 1))) else: - input_img = cv.resize(input_img, ((input_img.shape[0] + 1), input_img.shape[1])) + input_img = cv.resize(input_img, ((input_img.shape[0] + 1), input_img.shape[1])) elif input_img.shape[1] % 2 == 1: - input_img = cv.resize(input_img, (input_img.shape[0], input_img.shape[1] + 1)) + input_img = cv.resize(input_img, (input_img.shape[0], input_img.shape[1] + 1)) converted_input_img = cv.cvtColor(input_img, cv.COLOR_BGR2YUV_I420) elif args.output_image_format == 'GRAY': converted_input_img = cv.cvtColor(input_img, cv.COLOR_BGR2GRAY) @@ -192,43 +186,50 @@ def mkdir_output(args): return +def process(args, file_path): + if file_path.endswith(".txt"): + config = configparser.ConfigParser() + config.read(file_path) + input_node = json.loads(config['baseconf']['input_node']) + shape = json.loads(config['baseconf']['shape']) + input_node_np = np.array(input_node) + change_type_img_info = change_type(args, input_node_np) + img_info = np.reshape(change_type_img_info, shape) + out_path = os.path.join(args.output, os.path.splitext(os.path.split(file_path)[1])[0] + ".bin") + mkdir_output(args) + img_info.tofile(out_path) + else: + input_img = cv.imread(file_path) + if args.output_image_format == 'YUV': + resized_img1 = resize_img(args, input_img) + converted_img = convert_img(args, resized_img1) + mean_img = mean(args, converted_img) + else: + converted_img = convert_img(args, input_img) + resized_img = resize_img(args, converted_img) + mean_img = mean(args, resized_img) + coefficient_img = coefficient(args, mean_img) + change_type_img = change_type(args, coefficient_img) + change_format_img = change_format(args, change_type_img) + out_path = os.path.join(args.output, os.path.splitext(os.path.split(file_path)[1])[0] + ".bin") + mkdir_output(args) + change_format_img.tofile(out_path) + + def main(): """main function to receive params them change data to bin. """ args = get_args() - ret,is_dir = check_args(args) + ret, is_dir = check_args(args) if ret: if is_dir: - img_names = os.listdir(args.input) - for img_name in img_names: - img_path = os.path.join(args.input, img_name) - input_img = cv.imread(img_path) - if args.output_image_format == 'YUV': - resized_img1 = resize_img(args, input_img) - converted_img = convert_img(args, resized_img1) - mean_img = mean(args, converted_img) - else: - converted_img = convert_img(args, input_img) - resized_img = resize_img(args, converted_img) - mean_img = mean(args, resized_img) - coefficient_img = coefficient(args, mean_img) - change_type_img = change_type(args, coefficient_img) - change_format_img = change_format(args, change_type_img) - out_path = os.path.join(args.output, os.path.splitext(img_name)[0] + ".bin") - mkdir_output(args) - change_format_img.tofile(out_path) + files_name = os.listdir(args.input) + for file_name in files_name: + file_path = os.path.join(args.input, file_name) + process(args, file_path) else: - config = configparser.ConfigParser() - config.read(args.input) - input_node = json.loads(config['baseconf']['input_node']) - shape = json.loads(config['baseconf']['shape']) - input_node_np = np.array(input_node) - change_type_img_info = change_type(args, input_node_np) - img_info = np.reshape(change_type_img_info, shape) - out_path = os.path.join(args.output, os.path.splitext(args.input)[0] + ".bin") - mkdir_output(args) - img_info.tofile(out_path) - + process(args, args.input) + if __name__ == '__main__': main()