From 4d185a58588f9022eab7f5b593a12e039debc8cb Mon Sep 17 00:00:00 2001 From: Tony Lampada Date: Fri, 8 Aug 2025 12:39:48 +0000 Subject: [PATCH] add function to debug train api --- tests/manual/debugme.py | 33 ++++++++++++++++++++++++++++++++- 1 file changed, 32 insertions(+), 1 deletion(-) diff --git a/tests/manual/debugme.py b/tests/manual/debugme.py index b4d37941..77c6b929 100644 --- a/tests/manual/debugme.py +++ b/tests/manual/debugme.py @@ -5,6 +5,7 @@ os.environ["ROBOFLOW_CONFIG_DIR"] = f"{thisdir}/data/.config" from roboflow.roboflowpy import _argparser # noqa: E402 +from roboflow import Roboflow # import requests # requests.urllib3.disable_warnings() @@ -12,7 +13,8 @@ rootdir = os.path.abspath(f"{thisdir}/../..") sys.path.append(rootdir) -if __name__ == "__main__": + +def run_cli(): parser = _argparser() # args = parser.parse_args(["login"]) # args = parser.parse_args(f"upload {thisdir}/../datasets/chess -w wolfodorpythontests -p chess".split()) # noqa: E501 // docs @@ -45,3 +47,32 @@ # f"import -w tonyprivate -p meh-plvrv {thisdir}/../datasets/paligemma/".split() # noqa: E501 // docs ) args.func(args) + + +def run_api_train(): + rf = Roboflow() + project = rf.workspace("meh3").project("mosquitobao") + # version_number = project.generate_version( + # settings={ + # "augmentation": { + # "bbblur": {"pixels": 1.5}, + # "image": {"versions": 2}, + # }, + # "preprocessing": { + # "auto-orient": True, + # }, + # } + # ) + version_number = "61" + print(version_number) + version = project.version(version_number) + model = version.train( + speed="fast", # Options: "fast" (default) or "accurate" (paid feature) + checkpoint=None, # Use a specific checkpoint to continue training + ) + print(model) + + +if __name__ == "__main__": + # run_cli() + run_api_train()