image_to_gravity 源码

image_to_gravity 概述 该存储库提供了一个深层神经网络,该网络可以从一次射击中估算出重力方向。 数据集 一些数据集可在。 用法 以下命令仅是示例。 image_to_gravity / keep中提供了一些经过训练的模型。 回归 训练 $ cd *** /image_to_gravity/docker/docker $ ./run.sh $ cd regression $ python3 train.py 推理 $ cd *** /image_to_gravity/docker/docker $ ./run.sh $ cd regression $ python3 infer.