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
Loading
Loading
Loading
Loading
+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 ]

+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 }} 
+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
+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


+11.2 KiB
Loading image diff...
Loading