dk - edit

2026-08-20 16:44:27 +09:00
parent 2aee353809
commit c5156584a0
2 changed files with 13 additions and 1 deletions
@@ -1,6 +1,18 @@
+from serve_roformer import RoformerService
+
 def main():
    print("Hello from model-yolo-person-classify-market!")

+    # uv run python person_attr.py --image_url "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg" --xywh "404,290,74,193"
+
+    roformer_service = RoformerService()
+    roformer_service.load("net_last.pth")
+    result = roformer_service.predict({
+        "image_url": "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg",
+        "xywh": "404,290,74,193"
+    })
+    print(result)
+

 if __name__ == "__main__":
    main()