Docker镜像中默认传递模型权重文件路径的问题求助

问题根源 #

你遇到的问题有两个核心原因：

1. 参数名不匹配：代码中argparse定义的参数是--model-dir（短横线分隔），但Dockerfile的ENTRYPOINT里写的是--model_dir（下划线分隔），导致参数无法被正确解析。
2. ENTRYPOINT与CMD的组合逻辑错误：原配置中ENTRYPOINT包含了参数--model_dir，CMD仅提供参数值，当你在docker run后追加--image-path参数时，会直接覆盖CMD的内容，导致模型路径值丢失，最终命令变成python3 ./model.py --model_dir --image-path=happy.jpg，缺少模型路径的具体值。

解决方案 #

方案一：修正ENTRYPOINT与CMD的组合（推荐） #

调整Dockerfile的ENTRYPOINT和CMD结构，让ENTRYPOINT仅包含基础启动命令，CMD包含完整的默认模型参数，这样用户追加的参数会被追加到命令末尾，而非覆盖默认参数。

修改后的Dockerfile：

FROM python:3.10

WORKDIR /usr/src/app

RUN apt update && apt install -y python3-pip

COPY model.py model.pth.tar ./

ENTRYPOINT ["python3", "./model.py"]
CMD ["--model-dir", "./model.pth.tar"]

运行命令：

docker build -t my_model --progress=plain .

docker run my_model --image-path=happy.jpg

此时实际执行的命令为：python3 ./model.py --model-dir ./model.pth.tar --image-path=happy.jpg，两个参数都能被正确解析。

方案二：在代码中设置参数默认值 #

直接在model.py中给--model-dir设置默认值，无需在Dockerfile中传递该参数，更简洁直观。

修改后的model.py：

if __name__ == '__main__':
  parser = argparse.ArgumentParser()
  parser.add_argument('--image-path', type=str)
  parser.add_argument('--model-dir', type=str, default='./model.pth.tar')  # 添加默认路径
  args, _ = parser.parse_known_args()

  sample_image = args.image_path
  print("Sample:", sample_image)
  print("Model:", args.model_dir)
  print("Done.")

对应的Dockerfile可以简化为：

FROM python:3.10

WORKDIR /usr/src/app

RUN apt update && apt install -y python3-pip

COPY model.py model.pth.tar ./

ENTRYPOINT ["python3", "./model.py"]

运行命令：

docker build -t my_model --progress=plain .

docker run my_model --image-path=happy.jpg

代码会自动使用默认的./model.pth.tar作为模型路径，无需额外传递参数。

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