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

Merge pull request #145 from deepghs/dev/siglip 

dev(narugo): add generic support for siglip
parents be6a31db 415dbd60

Makefile

+6 −5
Original line number Diff line number Diff line
@@ -50,13 +50,14 @@ pdocs: 
dataset:

	mkdir -p ${DATASET_DIR}

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

		git clone https://huggingface.co/datasets/deepghs/chafen_arknights.git ${DATASET_DIR}/chafen_arknights; \

		hfutils download -r deepghs/chafen_arknights -t dataset -d . -o ${DATASET_DIR}/chafen_arknights; \

	fi

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

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

		hfutils download -r deepghs/monochrome_danbooru -t dataset -d . -o ${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; \

		hfutils download -r deepghs/character_similarity -t dataset -a images_test_v1.tar.xz -o ${DATASET_DIR}/images_test_v1; \

	fi

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

		hfutils download -r deepghs/realutils_unittest -a unsplash_1000.zip -o ${DATASET_DIR}/unsplash_1000; \

	fi

docs/source/api_doc/generic/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -11,4 +11,5 @@ imgutils.generic 


    classify

    enhance

    siglip

    yolo

docs/source/api_doc/generic/siglip.rst

0 → 100644
+35 −0
Original line number Diff line number Diff line 
imgutils.generic.siglip

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



.. currentmodule:: imgutils.generic.siglip



.. automodule:: imgutils.generic.siglip







SigLIPModel

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



.. autoclass:: SigLIPModel

    :members: __init__, image_encode, text_encode, predict, clear, make_ui, launch_demo





siglip_image_encode

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



.. autofunction:: siglip_image_encode





siglip_text_encode

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



.. autofunction:: siglip_text_encode





siglip_predict

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



.. autofunction:: siglip_predict

imgutils/generic/__init__.py

+1 −0
Original line number Diff line number Diff line
@@ -4,4 +4,5 @@ Overview: 
"""

from .classify import *

from .enhance import *

from .siglip import *

from .yolo import *

imgutils/generic/siglip.py

0 → 100644
+606 −0

File added.

Preview size limit exceeded, changes collapsed.