虽然PP-Humanv2包含了对摄像头的使用,但是针对不同用户,有些配置的输入的int,但是代码对int型没有进行识别。所以可能会出现以下报错:
针对该报错的解决办法是:
将下面代码
def set_file_name(self, path):
if path is not None:
self.file_name = os.path.split(path)[-1]
if "." in self.file_name:
self.file_name = self.file_name.split(".")[-2]
修改为
def set_file_name(self, path):
if type(path)==int:
self.file_name = path
elif path is not None:
self.file_name = os.path.split(path)[-1]
if "." in self.file_name:
self.file_name = self.file_name.split(".")[-2]
同时修改下面这段代码:
elif self.cfg['visual']:
video_out_name = 'output' if self.file_name is None else self.file_name
if "rtsp" in video_file:
video_out_name = video_out_name + "_t" + str(thread_idx).zfill(
2) + "_rtsp"
if not os.path.exists(self.output_dir):
改为:
elif self.cfg['visual']:
video_out_name = 'output' if (self.file_name is None or type(self.file_name)==int) else self.file_name
if type(video_file)==str and "rtsp" in video_file:
video_out_name = video_out_name + "_t" + str(thread_idx).zfill(
2) + "_rtsp"
if not os.path.exists(self.output_dir):
Over!