基于一些需求,我们现在需要将py文件转成exe的格式,但是可能会有一些路径参数需要灵活输入,所以需要在打包的时候设置好,
首先安装打包模块pyinstaller
pip install pyinstaller -i https://pypi.tuna.tsinghua.edu.cn/simple
一般都可以直接装上,装不上看具体报错,
这里补充一下:
在一开始装的时候可能会有路径问题找不到pyinstaller,
0.问题描述
报错内容为: pyinstaller : 无法将“pyinstaller”项识别为 cmdlet、函数、脚本文件或可运行程序的名称。请检查名称的拼写,如果包括路径,请确保路径正确,然后再试一次。所在位置 行:1
需要你去设置一下系统环境变量,这里我不想多写我直接贴一个地址你们去看,
基本上到这里你就都装上了,没有什么别的问题!
然后我们这里的问题是有需求设定参数,所以我们打包的命令是
pyinstaller -F .\split_train_val.py
后面的py文件找到对应的路径既可以
然后是调用命令,
.\split_train_val.exe --xml_path 'C:\\Users\\admin\\Desktop\\datasets\\VOCData\\Annotations\\' --txt_path 'C:\\Users\\admin\\Desktop\\datasets\\VOCData\\ImageSets\\Main\\'
重点来了,我们来看py文件的代码,
# coding:utf-8
import os
import random
import argparse
import sys
def split_train_val(input_path,output_path):
parser = argparse.ArgumentParser()
#xml文件的地址,根据自己的数据进行修改 xml一般存放在Annotations下
# 'Annotations\\'
# parser.add_argument('--xml_path', default='C:\\Users\\admin\\Desktop\\datasets\\VOCData\\Annotations\\', type=str, help='input xml label path')
parser.add_argument('--xml_path', default=input_path, type=str, help='input xml label path')
#数据集的划分,地址选择自己数据下的ImageSets/Main
# 'ImageSets\\Main\\'
# parser.add_argument('--txt_path', default='C:\\Users\\admin\\Desktop\\datasets\\VOCData\\ImageSets\\Main\\', type=str, help='output txt label path')
parser.add_argument('--txt_path', default=output_path, type=str, help='output txt label path')
opt = parser.parse_args()
trainval_percent = 1.0 # 训练集和验证集所占比例。 这里没有划分测试集
train_percent = 0.9 # 训练集所占比例,可自己进行调整
xmlfilepath = opt.xml_path
txtsavepath = opt.txt_path
total_xml = os.listdir(xmlfilepath)
if not os.path.exists(txtsavepath):
os.makedirs(txtsavepath)
num = len(total_xml)
list_index = range(num)
tv = int(num * trainval_percent)
tr = int(tv * train_percent)
trainval = random.sample(list_index, tv)
train = random.sample(trainval, tr)
file_trainval = open(txtsavepath + '/trainval.txt', 'w')
file_test = open(txtsavepath + '/test.txt', 'w')
file_train = open(txtsavepath + '/train.txt', 'w')
file_val = open(txtsavepath + '/val.txt', 'w')
for i in list_index:
name = total_xml[i][:-4] + '\n'
if i in trainval:
file_trainval.write(name)
if i in train:
file_train.write(name)
else:
file_val.write(name)
else:
file_test.write(name)
file_trainval.close()
file_train.close()
file_val.close()
file_test.close()
if __name__ == '__main__':
split_train_val(sys.argv[1], sys.argv[2])
其实这里重点需要看的只有下面这个部分。
parser = argparse.ArgumentParser()
#xml文件的地址,根据自己的数据进行修改 xml一般存放在Annotations下
# 'Annotations\\'
# parser.add_argument('--xml_path', default='C:\\Users\\admin\\Desktop\\datasets\\VOCData\\Annotations\\', type=str, help='input xml label path')
parser.add_argument('--xml_path', default=input_path, type=str, help='input xml label path')
#数据集的划分,地址选择自己数据下的ImageSets/Main
# 'ImageSets\\Main\\'
# parser.add_argument('--txt_path', default='C:\\Users\\admin\\Desktop\\datasets\\VOCData\\ImageSets\\Main\\', type=str, help='output txt label path')
parser.add_argument('--txt_path', default=output_path, type=str, help='output txt label path')
opt = parser.parse_args()
添加引用参数,所以我在调用的时候,是--xml_path 然后才是输入的路径,然后文件里面调用也很简单xmlfilepath = opt.xml_path。
到这里基本就解决我的出发点问题了,
敬礼!