Flower Pose Estimation Model

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

---

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)