dk - edit

2026-06-29 17:59:06 +09:00
parent a3bbf78ea2
commit 71b3c21f28
2 changed files with 50 additions and 174 deletions
@@ -19,10 +19,16 @@ transforms = T.Compose([
    T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
 ])

-task = Task.init(
-    project_name="Person_Attribute_Recognition", 
-    task_name="model-yolo-person-classify"
-)
+def init_acai_task(output):
+    task = Task.init(
+        project_name="Person_Attribute_Recognition", 
+        task_name="model-yolo-person-classify"
+    )
+
+    result_data = {"output": output, "status": "PASS"}
+    task.upload_artifact(name="final_result", artifact_object=result_data)
+
+


 class PredictDecoder(object):
@@ -103,8 +109,8 @@ def main(image, dataset='market', backbone='resnet50', use_id=False):
        print(f'{name}: {value}')
    print("=" * 50)

-    result_data = {"results": results, "status": "PASS"}
-    task.upload_artifact(name="final_result", artifact_object=result_data)
+    init_acai_task(results)
+


 if __name__ == "__main__":