Add YOLOv5 integration

README.md

@@ -152,6 +152,7 @@ The log, model checkpoints, optimizer states will be saved to the `exp` director
 
 - [Yikun Ji (Kunologist)](https://github.com/Gennadiyev)
 - [Qi Fan (fanqiNO1)](https://github.com/fanqiNO1)
+- [Xiangyun Rao (xyrrrrrrrr)](https://github.com/xyrrrrrrrr)
 
 ## Contributing
 

unipose/yolo/README.md

@@ -0,0 +1,19 @@
+# Yolo_unipose
+
++ main contributor: [Xiangyun Rao](https://github.com/xyrrrrrrrr)
+
+## Intro
+
+Yolo_unipose is a pre-processing tool for the [Unipose](https://github.com/Gennadiyev/unipose), which is based on the [YOLOv5](https://github.com/ultralytics/yolov5). It is used to get the main creature in the image, judge its kind, and then return the bounding box of the creature.
+
+## How to use
+
+If you want to see how it works, you can run the `yolo_unipose.py` file directly. The default image is `rxy.jpg`. You can also change the image path in the code or in the command line.
+
+```bash
+python yolo_unipose.py --source your_image_path
+```
+
+## Reference
+
++ [YOLOv5](https://github.com/ultralytics/yolov5)

unipose/yolo/__init__.py

@@ -0,0 +1,8 @@
+"""Implements UniPose model.
+@author: Xiangyun Rao
+"""
+
+from .yolo_unipose import yolo_unipose
+
+
+__all__ = ["yolo_unipose"]