Docker环境运行LipNet数据管道时遇OpenCV及文件不存在错误

错误原因 #

你的错误核心是文件名解析失败，导致生成了空的文件名，进而找不到对应的.mpg和.align文件：

• 代码中为Windows环境写了path.split('\\')来分割路径，但Docker容器是Linux环境，路径分隔符为/，导致split('\\')无法正确拆分出文件名。
• 后续用split('.')[0]处理时，因为路径以./开头，会得到空字符串，最终生成data/s1/.mpg和data/alignments/s1/.align这类无效路径，触发文件找不到的错误。
• OpenCV把data/s1/.mpg当成了图片序列模板（而非视频文件），因此抛出CAP_IMAGES: can't find starting number的错误。

解决方案 #

1. 替换跨平台兼容的文件名解析代码 #

不要手动用分隔符拆分路径，改用Python标准库的os.path模块来处理，它会自动适配Windows/Linux的路径规则：

修改load_data函数中的文件名解析部分：

def load_data(path: str): 
    path = bytes.decode(path.numpy())
    # 替换原有的手动split逻辑，跨平台获取不带扩展名的文件名
    file_name = os.path.splitext(os.path.basename(path))[0]
    video_path = os.path.join('data','s1',f'{file_name}.mpg')
    alignment_path = os.path.join('data','alignments','s1',f'{file_name}.align')
    frames = load_video(video_path) 
    alignments = load_alignments(alignment_path)
    
    return frames, alignments

2. 确保挂载路径与文件结构正确 #

检查宿主机的文件结构：

• 宿主机的./lipnet目录下必须包含data文件夹，结构如下：

  lipnet/
  ├─ data/
  │  ├─ s1/
  │  │  └─ [多个.mpg视频文件]
  │  └─ alignments/
  │     └─ s1/
  │        └─ [对应视频的.align文件]
  ├─ Dockerfile
  └─ docker-compose.yaml
  

• Docker挂载的./lipnet:/lipnet会将宿主机的lipnet目录映射到容器的/lipnet工作目录，确保容器内能访问到/lipnet/data下的文件。

3. 可选：为tf.py_function返回值设置形状 #

tf.py_function不会自动推断返回张量的形状，可能导致后续操作出错，建议在mappable_function中添加形状设置（根据你的实际数据形状调整）：

def mappable_function(path:str):
    frames, alignments = tf.py_function(load_data, [path], (tf.float32, tf.int64))
    # 假设frames是(帧数, 高度, 宽度, 通道数)，alignments是(帧数,)
    frames.set_shape((None, None, None, 3))
    alignments.set_shape((None,))
    return frames, alignments

验证 #

修改后重新启动Docker容器，运行data.as_numpy_iterator().next()，应该能正确加载视频帧和对齐文件。

内容的提问来源于stack exchange，提问作者Swapnil Kumar