我可以用 PyTorch 训练我的深度神经网络,然后打包成 VisionCapsule 吗?BrainFrame 支持 PyTorch 框架吗?
可以,你可以用 PyTorch 训练你的深度神经网络,并把它打包成 VisionCapsule。BrainFrame 默认包含 TensorFlow、OpenCV DNN 和 OpenVINO 支持,但如果你使用 PyTorch 或其他默认不包含的框架,需要执行以下步骤,才能让你的 VisionCapsule 在 BrainFrame 中工作:
- 启动 BrainFrame core docker 容器,并安装默认不包含的 PyTorch(或其他)所需模块。可以执行以下命令来完成:
docker exec -it brainframe_core_1 /bin/bash
pip install torchvision==0.11.2
pip install pandas==1.1.5
exit
- 找到 brainframe_core_1 docker 容器的 ID,并用下面的命令提交更新后的 BrainFrame core 镜像:
docker container ls
docker commit --pause abd9f29a25c1 brainframe_core:0.29.6_new
- 创建一个 docker-compose.override.yml 文件,以便下次启动 BrainFrame 时使用 brainframe_core:0.29.6_new docker 镜像。文件内容应为:
version: '2.4'
services:
core:
image: brainframe_core:0.29.6_new
把 docker-compose.override.yml 文件保存在 BrainFrame 的 install_path 中,通常是 /usr/local/share/brainframe/。你可以通过运行下面的命令找到 BrainFrame 的 install_path:
brainframe info