花の姿勢推定モデル

Given a flower image cropped from an object detection bounding box, this model predicts its pose in Euler angles.

• https://github.com/AgriSwarm/pose_estimate_poc

---

Setup:

cd mlcore/sixdrepnet
pip install poetry
poetry install

Train:

poetry run python main.py

Predict:

poetry run python inference.py

Or in Python:

from inference import predict
results = predict("path/to/images")
print(results)