Unverified Commit 44b2ef2c authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #17 from deepghs/ccip 

dev(7eu7d): ccip, caihongmaoniubi, @7eu7d7
parents 2819e994 925a3a9b

.github/workflows/doc.yml

+1 −1
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ name: Docs Deploy 


on:

  push:

    branches: [ main, 'doc/*', 'dev/*' ]

  #    branches: [ main, 'doc/*', 'dev/*' ]

  release:

    types: [ published ]

.github/workflows/export.yml

+17 −1
Original line number Diff line number Diff line
@@ -19,8 +19,9 @@ jobs: 
          #          - 'lpips'

          #          - 'monochrome'

          #          - 'person_detect'

          - 'face_detect'

          #          - 'face_detect'

          #          - 'manbits_detect'

          - 'ccip'



    steps:

      - name: Checkout code
@@ -36,7 +37,22 @@ jobs: 
          pip install --upgrade pip

          pip install --upgrade flake8 setuptools wheel twine

          pip install -r requirements.txt

          pip install -r requirements-test.txt

          pip install 'torch<2'

          pip install -r requirements-zoo.txt

      - name: Prepare dataset

        uses: nick-fields/retry@v2

        env:

          CI: 'true'

          HF_NARUGO_USERNAME: ${{ secrets.HF_NARUGO_USERNAME }}

          HF_NARUGO_PASSWORD: ${{ secrets.HF_NARUGO_PASSWORD }}

        with:

          shell: bash

          timeout_minutes: 20

          max_attempts: 5

          retry_on: any

          command: |

            make dataset

      - name: Transform models to onnx format

        run: |

          python -m zoo.${{ matrix.model-name }} export -O ./${{ matrix.model-name }}

Makefile

+6 −1
Original line number Diff line number Diff line
@@ -55,3 +55,8 @@ dataset: 
	if [ ! -d ${DATASET_DIR}/monochrome_danbooru ]; then \

		git clone https://${HF_NARUGO_USERNAME}:${HF_NARUGO_PASSWORD}@huggingface.co/datasets/deepghs/monochrome_danbooru.git ${DATASET_DIR}/monochrome_danbooru; \

	fi

	if [ ! -d ${DATASET_DIR}/images_test_v1 ]; then \

		mkdir -p ${DATASET_DIR}/images_test_v1 && \

		curl -L -o ${DATASET_DIR}/images_test_v1/images_test_v1.tar.xz https://huggingface.co/datasets/deepghs/character_similarity/resolve/main/images_test_v1.tar.xz && \

		cd ${DATASET_DIR}/images_test_v1 && tar -xvf images_test_v1.tar.xz && rm -rf *.tar.xz; \

	fi

docs/source/api_doc/metrics/ccip.rst

0 → 100644
+71 −0
Original line number Diff line number Diff line 
imgutils.metrics.ccip

===========================



.. currentmodule:: imgutils.metrics.ccip



.. automodule:: imgutils.metrics.ccip





ccip_extract_feature

--------------------------------------------



.. autofunction:: ccip_extract_feature







ccip_batch_extract_features

--------------------------------------------



.. autofunction:: ccip_batch_extract_features







ccip_default_threshold

--------------------------------------------



.. autofunction:: ccip_default_threshold







ccip_difference

--------------------------------------------



.. autofunction:: ccip_difference







ccip_same

--------------------------------------------



.. autofunction:: ccip_same







ccip_batch_differences

--------------------------------------------



.. autofunction:: ccip_batch_differences







ccip_batch_same

--------------------------------------------



.. autofunction:: ccip_batch_same







ccip_default_clustering_params

--------------------------------------------



.. autofunction:: ccip_default_clustering_params







ccip_clustering

--------------------------------------------



.. autofunction:: ccip_clustering

docs/source/api_doc/metrics/ccip/1.jpg

0 → 100644
+11.2 KiB
Loading image diff...