From 9aee0557e545330296a7be7ba6f7f10efed71b13 Mon Sep 17 00:00:00 2001 From: SWHL Date: Tue, 23 Apr 2024 21:19:19 +0800 Subject: [PATCH] Add CLI of yolov8 convert --- .github/workflows/gen_whl_to_pypi.yml | 2 +- setup.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/gen_whl_to_pypi.yml b/.github/workflows/gen_whl_to_pypi.yml index 01e6361..4177aa2 100644 --- a/.github/workflows/gen_whl_to_pypi.yml +++ b/.github/workflows/gen_whl_to_pypi.yml @@ -51,7 +51,7 @@ jobs: pip install -r requirements.txt python -m pip install --upgrade pip pip install wheel get_pypi_latest_version - python setup.py bdist_wheel ${{ github.event.head_commit.message }} + python setup.py bdist_wheel ${{ github.ref_name }} # - name: Publish distribution 📦 to Test PyPI # uses: pypa/gh-action-pypi-publish@v1.5.0 diff --git a/setup.py b/setup.py index e3f0552..04c1b0d 100644 --- a/setup.py +++ b/setup.py @@ -82,6 +82,8 @@ def get_readme() -> str: f"yolov5_yaml_to_coco={MODULE_NAME}.yolov5_yaml_to_coco:main", f"labelImg_to_publaynet={MODULE_NAME}.labelImg_to_publaynet:main", f"labelme_to_coco={MODULE_NAME}.labelme_to_coco:main", + f"yolov5_to_yolov8={MODULE_NAME}.yolov5_to_yolov8:main", + f"yolov8_to_yolov5={MODULE_NAME}.yolov8_to_yolov5:main", ], }, )