Browse Source

merge conflict

master
Yingda Chen 2 years ago
parent
commit
d3c548fe14
100 changed files with 1753 additions and 0 deletions
  1. +8
    -0
      .dev_scripts/build_docs.sh
  2. +169
    -0
      .dev_scripts/build_image.sh
  3. +36
    -0
      .dev_scripts/ci_container_test.sh
  4. +80
    -0
      .dev_scripts/dockerci.sh
  5. +3
    -0
      .dev_scripts/linter.sh
  6. +7
    -0
      .dev_scripts/run_docker.sh
  7. +11
    -0
      .dockerignore
  8. +9
    -0
      .gitattributes
  9. +64
    -0
      .github/workflows/citest.yaml
  10. +22
    -0
      .github/workflows/lint.yaml
  11. +128
    -0
      .gitignore
  12. +37
    -0
      .pre-commit-config.yaml
  13. +37
    -0
      .pre-commit-config_local.yaml
  14. +28
    -0
      .readthedocs.yaml
  15. +204
    -0
      LICENSE
  16. +1
    -0
      MANIFEST.in
  17. +25
    -0
      Makefile
  18. +67
    -0
      Makefile.docker
  19. +27
    -0
      README.md
  20. +1
    -0
      configs/README.md
  21. +178
    -0
      configs/cv/configuration.json
  22. +7
    -0
      configs/examples/configuration.json
  23. +2
    -0
      configs/examples/configuration.py
  24. +4
    -0
      configs/examples/configuration.yaml
  25. +5
    -0
      configs/examples/plain_args.yaml
  26. +131
    -0
      configs/examples/train.json
  27. +87
    -0
      configs/nlp/sbert_sentence_similarity.json
  28. +62
    -0
      configs/nlp/sequence_classification_trainer.yaml
  29. +3
    -0
      data/test/audios/1ch_nihaomiya.wav
  30. +3
    -0
      data/test/audios/3ch_nihaomiya.wav
  31. +3
    -0
      data/test/audios/asr_example.wav
  32. +3
    -0
      data/test/audios/asr_example_8K.wav
  33. +3
    -0
      data/test/audios/asr_example_cn_dialect.wav
  34. +3
    -0
      data/test/audios/asr_example_cn_en.wav
  35. +3
    -0
      data/test/audios/asr_example_en.wav
  36. +3
    -0
      data/test/audios/asr_example_es.wav
  37. +3
    -0
      data/test/audios/asr_example_id.wav
  38. +3
    -0
      data/test/audios/asr_example_ja.wav
  39. +3
    -0
      data/test/audios/asr_example_ko.wav
  40. +3
    -0
      data/test/audios/asr_example_ru.wav
  41. +3
    -0
      data/test/audios/farend_speech.wav
  42. +3
    -0
      data/test/audios/kws_bofangyinyue.wav
  43. +3
    -0
      data/test/audios/kws_xiaoyunxiaoyun.wav
  44. +3
    -0
      data/test/audios/nearend_mic.wav
  45. +3
    -0
      data/test/audios/noise_2ch.wav
  46. +3
    -0
      data/test/audios/speech_with_noise.wav
  47. +3
    -0
      data/test/audios/wake_word_with_label_xyxy.wav
  48. +3
    -0
      data/test/images/Solvay_conference_1927.png
  49. +3
    -0
      data/test/images/auto_demo.jpg
  50. +3
    -0
      data/test/images/bird.JPEG
  51. +3
    -0
      data/test/images/card_detection.jpg
  52. +51
    -0
      data/test/images/coco_cn/train.json
  53. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000021183.jpg
  54. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000177625.jpg
  55. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000197406.jpg
  56. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000275612.jpg
  57. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000404748.jpg
  58. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000493952.jpg
  59. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000496606.jpg
  60. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000563734.jpg
  61. +3
    -0
      data/test/images/coco_cn/train/COCO_train2014_000000573854.jpg
  62. +52
    -0
      data/test/images/coco_cn/val.json
  63. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000043734.jpg
  64. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000044723.jpg
  65. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000163020.jpg
  66. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000341725.jpg
  67. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000412975.jpg
  68. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000473869.jpg
  69. +3
    -0
      data/test/images/coco_cn/val/COCO_val2014_000000574392.jpg
  70. +3
    -0
      data/test/images/crowd_counting.jpg
  71. +3
    -0
      data/test/images/dogs.jpg
  72. +3
    -0
      data/test/images/face_detection.png
  73. +3
    -0
      data/test/images/face_detection2.jpeg
  74. +3
    -0
      data/test/images/face_emotion.jpg
  75. +3
    -0
      data/test/images/face_enhancement/gt/000000.jpg
  76. +3
    -0
      data/test/images/face_enhancement/gt/000001.jpg
  77. +3
    -0
      data/test/images/face_enhancement/lq/000000.png
  78. +3
    -0
      data/test/images/face_enhancement/lq/000001.png
  79. +3
    -0
      data/test/images/face_human_hand_detection.jpg
  80. +3
    -0
      data/test/images/face_recognition_1.png
  81. +3
    -0
      data/test/images/face_recognition_2.png
  82. +3
    -0
      data/test/images/facial_expression_recognition.jpg
  83. +3
    -0
      data/test/images/generative_multimodal.jpg
  84. +3
    -0
      data/test/images/hand_keypoints.jpg
  85. +3
    -0
      data/test/images/hand_static.jpg
  86. +3
    -0
      data/test/images/image-text-retrieval.jpg
  87. +3
    -0
      data/test/images/image_body_reshaping.jpg
  88. +3
    -0
      data/test/images/image_captioning.png
  89. +3
    -0
      data/test/images/image_classification.png
  90. +3
    -0
      data/test/images/image_color_enhance.png
  91. +3
    -0
      data/test/images/image_color_enhance/gt/1.png
  92. +3
    -0
      data/test/images/image_color_enhance/gt/2.png
  93. +3
    -0
      data/test/images/image_color_enhance/gt/3.png
  94. +3
    -0
      data/test/images/image_color_enhance/gt/4.png
  95. +3
    -0
      data/test/images/image_color_enhance/lq/1.png
  96. +3
    -0
      data/test/images/image_color_enhance/lq/2.png
  97. +3
    -0
      data/test/images/image_color_enhance/lq/3.png
  98. +3
    -0
      data/test/images/image_color_enhance/lq/4.png
  99. +3
    -0
      data/test/images/image_detection.jpg
  100. +3
    -0
      data/test/images/image_inpainting/image_inpainting.png

+ 8
- 0
.dev_scripts/build_docs.sh View File

@@ -0,0 +1,8 @@
pip install -r requirements/docs.txt
cd docs
rm -rf build

# update api rst
#rm -rf source/api/
#sphinx-apidoc --module-first -o source/api/ ../modelscope/
make html

+ 169
- 0
.dev_scripts/build_image.sh View File

@@ -0,0 +1,169 @@
#!/bin/bash
# default values.
BASE_CPU_IMAGE=reg.docker.alibaba-inc.com/modelscope/ubuntu:20.04
BASE_GPU_IMAGE=reg.docker.alibaba-inc.com/modelscope/ubuntu:20.04-cuda11.3.0-cudnn8-devel
MODELSCOPE_REPO_ADDRESS=reg.docker.alibaba-inc.com/modelscope/modelscope
python_version=3.7.13
torch_version=1.11.0
cudatoolkit_version=11.3
tensorflow_version=1.15.5
modelscope_version=None
is_ci_test=False
is_dsw=False
is_cpu=False
run_ci_test=False
function usage(){
echo "usage: build.sh "
echo " --python=python_version set python version, default: $python_version"
echo " --torch=torch_version set pytorch version, fefault: $torch_version"
echo " --cudatoolkit=cudatoolkit_version set cudatoolkit version used for pytorch, default: $cudatoolkit_version"
echo " --tensorflow=tensorflow_version set tensorflow version, default: $tensorflow_version"
echo " --modelscope=modelscope_version set modelscope version, default: $modelscope_version"
echo " --test option for run test before push image, only push on ci test pass"
echo " --cpu option for build cpu version"
echo " --dsw option for build dsw version"
echo " --ci option for build ci version"
echo " --push option for push image to remote repo"
}
for i in "$@"; do
case $i in
--python=*)
python_version="${i#*=}"
shift
;;
--torch=*)
torch_version="${i#*=}"
shift # pytorch version
;;
--tensorflow=*)
tensorflow_version="${i#*=}"
shift # tensorflow version
;;
--cudatoolkit=*)
cudatoolkit_version="${i#*=}"
shift # cudatoolkit for pytorch
;;
--modelscope=*)
modelscope_version="${i#*=}"
shift # cudatoolkit for pytorch
;;
--test)
run_ci_test=True
shift # will run ci test
;;
--cpu)
is_cpu=True
shift # is cpu image
;;
--ci)
is_ci_test=True
shift # is ci, will not install modelscope
;;
--dsw)
is_dsw=True
shift # is dsw, will set dsw cache location
;;
--push)
is_push=True
shift # is dsw, will set dsw cache location
;;
--help)
usage
exit 0
;;
-*|--*)
echo "Unknown option $i"
usage
exit 1
;;
*)
;;
esac
done

if [ "$modelscope_version" == "None" ]; then
echo "ModelScope version must specify!"
exit 1
fi
if [ "$is_cpu" == "True" ]; then
export BASE_IMAGE=$BASE_CPU_IMAGE
base_tag=ubuntu20.04
export USE_GPU=False
else
export BASE_IMAGE=$BASE_GPU_IMAGE
base_tag=ubuntu20.04-cuda11.3.0
export USE_GPU=True
fi
if [[ $python_version == 3.7* ]]; then
base_tag=$base_tag-py37
elif [[ $python_version == z* ]]; then
base_tag=$base_tag-py38
elif [[ $python_version == z* ]]; then
base_tag=$base_tag-py39
else
echo "Unsupport python version: $python_version"
exit 1
fi

target_image_tag=$base_tag-torch$torch_version-tf$tensorflow_version
if [ "$is_ci_test" == "True" ]; then
target_image_tag=$target_image_tag-$modelscope_version-ci
else
target_image_tag=$target_image_tag-$modelscope_version-test
fi
export IMAGE_TO_BUILD=$MODELSCOPE_REPO_ADDRESS:$target_image_tag
export PYTHON_VERSION=$python_version
export TORCH_VERSION=$torch_version
export CUDATOOLKIT_VERSION=$cudatoolkit_version
export TENSORFLOW_VERSION=$tensorflow_version
echo -e "Building image with:\npython$python_version\npytorch$torch_version\ntensorflow:$tensorflow_version\ncudatoolkit:$cudatoolkit_version\ncpu:$is_cpu\nis_ci:$is_ci_test\nis_dsw:$is_dsw\n"
docker_file_content=`cat docker/Dockerfile.ubuntu`
if [ "$is_ci_test" != "True" ]; then
echo "Building ModelScope lib, will install ModelScope lib to image"
docker_file_content="${docker_file_content} \nRUN pip install --no-cache-dir modelscope==$modelscope_version -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html"
fi
echo "$is_dsw"
if [ "$is_dsw" == "False" ]; then
echo "Not DSW image"
else
echo "Building dsw image well need set ModelScope lib cache location."
docker_file_content="${docker_file_content} \nENV MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope"
fi
printf "$docker_file_content" > Dockerfile
docker build -t $IMAGE_TO_BUILD \
--build-arg USE_GPU \
--build-arg BASE_IMAGE \
--build-arg PYTHON_VERSION \
--build-arg TORCH_VERSION \
--build-arg CUDATOOLKIT_VERSION \
--build-arg TENSORFLOW_VERSION \
-f Dockerfile .

if [ $? -ne 0 ]; then
echo "Running docker build command error, please check the log!"
exit -1
fi
if [ "$run_ci_test" == "True" ]; then
echo "Running ci case."
export MODELSCOPE_CACHE=/home/mulin.lyh/model_scope_cache
export MODELSCOPE_HOME_CACHE=/home/mulin.lyh/ci_case_home # for credential
export IMAGE_NAME=$MODELSCOPE_REPO_ADDRESS
export IMAGE_VERSION=$target_image_tag
export MODELSCOPE_DOMAIN=www.modelscope.cn
export HUB_DATASET_ENDPOINT=http://www.modelscope.cn
export CI_TEST=True
export TEST_LEVEL=1
if [ "$is_ci_test" != "True" ]; then
echo "Testing for dsw image or MaaS-lib image"
export CI_COMMAND="python tests/run.py"
fi
bash .dev_scripts/dockerci.sh
if [ $? -ne 0 ]; then
echo "Running unittest failed, please check the log!"
exit -1
fi
fi
if [ "$is_push" == "True" ]; then
echo "Pushing image: $IMAGE_TO_BUILD"
docker push $IMAGE_TO_BUILD
fi

+ 36
- 0
.dev_scripts/ci_container_test.sh View File

@@ -0,0 +1,36 @@
echo "Testing envs"
printenv
echo "ENV END"
if [ "$MODELSCOPE_SDK_DEBUG" == "True" ]; then
awk -F: '/^[^#]/ { print $1 }' requirements/framework.txt | xargs -n 1 pip install -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
awk -F: '/^[^#]/ { print $1 }' requirements/audio.txt | xargs -n 1 pip install -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
awk -F: '/^[^#]/ { print $1 }' requirements/cv.txt | xargs -n 1 pip install -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
awk -F: '/^[^#]/ { print $1 }' requirements/multi-modal.txt | xargs -n 1 pip install -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
awk -F: '/^[^#]/ { print $1 }' requirements/nlp.txt | xargs -n 1 pip install -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
pip install -r requirements/tests.txt

git config --global --add safe.directory /Maas-lib
git config --global user.email tmp
git config --global user.name tmp.com

# linter test
# use internal project for pre-commit due to the network problem
if [ `git remote -v | grep alibaba | wc -l` -gt 1 ]; then
pre-commit run -c .pre-commit-config_local.yaml --all-files
if [ $? -ne 0 ]; then
echo "linter test failed, please run 'pre-commit run --all-files' to check"
exit -1
fi
fi
# test with install
python setup.py install
else
echo "Running case in release image, run case directly!"
fi
if [ $# -eq 0 ]; then
ci_command="python tests/run.py --subprocess"
else
ci_command="$@"
fi
echo "Running case with command: $ci_command"
$ci_command

+ 80
- 0
.dev_scripts/dockerci.sh View File

@@ -0,0 +1,80 @@
#!/bin/bash
MODELSCOPE_CACHE_DIR_IN_CONTAINER=/modelscope_cache
CODE_DIR=$PWD
CODE_DIR_IN_CONTAINER=/Maas-lib
echo "$USER"
gpus='7 6 5 4 3 2 1 0'
cpu_sets='0-7 8-15 16-23 24-30 31-37 38-44 45-51 52-58'
cpu_sets_arr=($cpu_sets)
is_get_file_lock=false
# export RUN_CASE_COMMAND='python tests/run.py --run_config tests/run_config.yaml'
CI_COMMAND=${CI_COMMAND:-bash .dev_scripts/ci_container_test.sh $RUN_CASE_BASE_COMMAND}
echo "ci command: $CI_COMMAND"
for gpu in $gpus
do
exec {lock_fd}>"/tmp/gpu$gpu" || exit 1
flock -n "$lock_fd" || { echo "WARN: gpu $gpu is in use!" >&2; continue; }
echo "get gpu lock $gpu"
CONTAINER_NAME="modelscope-ci-$gpu"
let is_get_file_lock=true

# pull image if there are update
docker pull ${IMAGE_NAME}:${IMAGE_VERSION}
if [ "$MODELSCOPE_SDK_DEBUG" == "True" ]; then
docker run --rm --name $CONTAINER_NAME --shm-size=16gb \
--cpuset-cpus=${cpu_sets_arr[$gpu]} \
--gpus="device=$gpu" \
-v $CODE_DIR:$CODE_DIR_IN_CONTAINER \
-v $MODELSCOPE_CACHE:$MODELSCOPE_CACHE_DIR_IN_CONTAINER \
-v $MODELSCOPE_HOME_CACHE/$gpu:/root \
-v /home/admin/pre-commit:/home/admin/pre-commit \
-e CI_TEST=True \
-e TEST_LEVEL=$TEST_LEVEL \
-e MODELSCOPE_CACHE=$MODELSCOPE_CACHE_DIR_IN_CONTAINER \
-e MODELSCOPE_DOMAIN=$MODELSCOPE_DOMAIN \
-e MODELSCOPE_SDK_DEBUG=True \
-e HUB_DATASET_ENDPOINT=$HUB_DATASET_ENDPOINT \
-e TEST_ACCESS_TOKEN_CITEST=$TEST_ACCESS_TOKEN_CITEST \
-e TEST_ACCESS_TOKEN_SDKDEV=$TEST_ACCESS_TOKEN_SDKDEV \
-e TEST_LEVEL=$TEST_LEVEL \
-e MODELSCOPE_ENVIRONMENT='ci' \
-e TEST_UPLOAD_MS_TOKEN=$TEST_UPLOAD_MS_TOKEN \
-e MODEL_TAG_URL=$MODEL_TAG_URL \
--workdir=$CODE_DIR_IN_CONTAINER \
--net host \
${IMAGE_NAME}:${IMAGE_VERSION} \
$CI_COMMAND
else
docker run --rm --name $CONTAINER_NAME --shm-size=16gb \
--cpuset-cpus=${cpu_sets_arr[$gpu]} \
--gpus="device=$gpu" \
-v $CODE_DIR:$CODE_DIR_IN_CONTAINER \
-v $MODELSCOPE_CACHE:$MODELSCOPE_CACHE_DIR_IN_CONTAINER \
-v $MODELSCOPE_HOME_CACHE/$gpu:/root \
-v /home/admin/pre-commit:/home/admin/pre-commit \
-e CI_TEST=True \
-e TEST_LEVEL=$TEST_LEVEL \
-e MODELSCOPE_CACHE=$MODELSCOPE_CACHE_DIR_IN_CONTAINER \
-e MODELSCOPE_DOMAIN=$MODELSCOPE_DOMAIN \
-e HUB_DATASET_ENDPOINT=$HUB_DATASET_ENDPOINT \
-e TEST_ACCESS_TOKEN_CITEST=$TEST_ACCESS_TOKEN_CITEST \
-e TEST_ACCESS_TOKEN_SDKDEV=$TEST_ACCESS_TOKEN_SDKDEV \
-e TEST_LEVEL=$TEST_LEVEL \
-e MODELSCOPE_ENVIRONMENT='ci' \
-e TEST_UPLOAD_MS_TOKEN=$TEST_UPLOAD_MS_TOKEN \
-e MODEL_TAG_URL=$MODEL_TAG_URL \
--workdir=$CODE_DIR_IN_CONTAINER \
--net host \
${IMAGE_NAME}:${IMAGE_VERSION} \
$CI_COMMAND
fi
if [ $? -ne 0 ]; then
echo "Running test case failed, please check the log!"
exit -1
fi
break
done
if [ "$is_get_file_lock" = false ] ; then
echo 'No free GPU!'
exit 1
fi

+ 3
- 0
.dev_scripts/linter.sh View File

@@ -0,0 +1,3 @@
yapf -r -i modelscope/ configs/ tests/ setup.py
isort -rc modelscope/ configs/ tests/ setup.py
flake8 modelscope/ configs/ tests/ setup.py

+ 7
- 0
.dev_scripts/run_docker.sh View File

@@ -0,0 +1,7 @@
#sudo docker run --name zwm_maas -v /home/wenmeng.zwm/workspace:/home/wenmeng.zwm/workspace --net host -ti reg.docker.alibaba-inc.com/pai-dlc/tensorflow-training:2.3-gpu-py36-cu101-ubuntu18.04 bash
#sudo docker run --name zwm_maas_pytorch -v /home/wenmeng.zwm/workspace:/home/wenmeng.zwm/workspace --net host -ti reg.docker.alibaba-inc.com/pai-dlc/pytorch-training:1.10PAI-gpu-py36-cu113-ubuntu18.04 bash
CONTAINER_NAME=modelscope-dev
IMAGE_NAME=registry.cn-shanghai.aliyuncs.com/modelscope/modelscope
IMAGE_VERSION=v0.1.1-16-g62856fa-devel
MOUNT_DIR=/home/wenmeng.zwm/workspace
sudo docker run --name $CONTAINER_NAME -v $MOUNT_DIR:$MOUNT_DIR --net host -ti ${IMAGE_NAME}:${IMAGE_VERSION} bash

+ 11
- 0
.dockerignore View File

@@ -0,0 +1,11 @@
.gitignore
tests
data
.dev_scripts
.dockerignore
.git
.gitattributes
.pre-commit-config.yaml
.pre-commit-config_local.yaml
.readthedocs.yaml
Dockfile

+ 9
- 0
.gitattributes View File

@@ -0,0 +1,9 @@
*.png filter=lfs diff=lfs merge=lfs -text
*.jpg filter=lfs diff=lfs merge=lfs -text
*.mp4 filter=lfs diff=lfs merge=lfs -text
*.wav filter=lfs diff=lfs merge=lfs -text
*.JPEG filter=lfs diff=lfs merge=lfs -text
*.jpeg filter=lfs diff=lfs merge=lfs -text
*.pickle filter=lfs diff=lfs merge=lfs -text
*.avi filter=lfs diff=lfs merge=lfs -text
*.bin filter=lfs diff=lfs merge=lfs -text

+ 64
- 0
.github/workflows/citest.yaml View File

@@ -0,0 +1,64 @@
name: citest

on:
push:
branches:
- master
- "release/**"
paths-ignore:
- "setup.*"
- "requirements.txt"
- "requirements/**"
- "docs/**"
- "tools/**"
- ".dev_scripts/**"
- "README.md"
- "README_zh-CN.md"
- "NOTICE"
- ".github/workflows/lint.yaml"
- ".github/workflows/publish.yaml"

pull_request:
paths-ignore:
- "setup.*"
- "requirements.txt"
- "requirements/**"
- "docs/**"
- "tools/**"
- ".dev_scripts/**"
- "README.md"
- "README_zh-CN.md"
- "NOTICE"
- ".github/workflows/lint.yaml"
- ".github/workflows/publish.yaml"

concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true

jobs:
unittest:
# The type of runner that the job will run on
runs-on: [modelscope-self-hosted]
steps:
- name: ResetFileMode
shell: bash
run: |
# reset filemode to allow action runner to delete files
# generated by root in docker
set -e
source ~/.bashrc
sudo chown -R $USER:$USER $ACTION_RUNNER_DIR

- name: Checkout
uses: actions/checkout@v2
with:
lfs: 'true'
- name: Checkout LFS objects
run: git lfs checkout
- name: Run unittest
shell: bash
run: |
set -e
source /mnt/modelscope/ci_env.sh
bash .dev_scripts/dockerci.sh

+ 22
- 0
.github/workflows/lint.yaml View File

@@ -0,0 +1,22 @@
name: Lint test

on: [push, pull_request]

concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true

jobs:
lint:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- name: Set up Python 3.7
uses: actions/setup-python@v2
with:
python-version: 3.7
- name: Install pre-commit hook
run: |
pip install pre-commit
- name: Linting
run: pre-commit run --all-files

+ 128
- 0
.gitignore View File

@@ -0,0 +1,128 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
*.egg-info/
.installed.cfg
*.egg
/package
/temp
MANIFEST

# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
.hypothesis/
.pytest_cache/

# Translations
*.mo
*.pot

# Django stuff:
*.log
local_settings.py
db.sqlite3

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
target/

# Jupyter Notebook
.ipynb_checkpoints

# pyenv
.python-version

# celery beat schedule file
celerybeat-schedule

# SageMath parsed files
*.sage.py

# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/

.vscode
.idea

# custom
*.pkl
*.pkl.json
*.log.json
*.whl
*.tar.gz
*.swp
*.log
*.tar.gz
source.sh
tensorboard.sh
.DS_Store
replace.sh
result.png

# Pytorch
*.pth
*.pt

+ 37
- 0
.pre-commit-config.yaml View File

@@ -0,0 +1,37 @@
repos:
- repo: https://gitlab.com/pycqa/flake8.git
rev: 4.0.0
hooks:
- id: flake8
exclude: thirdparty/|examples/
- repo: https://github.com/PyCQA/isort.git
rev: 4.3.21
hooks:
- id: isort
exclude: examples
- repo: https://github.com/pre-commit/mirrors-yapf.git
rev: v0.30.0
hooks:
- id: yapf
exclude: thirdparty/|examples/
- repo: https://github.com/pre-commit/pre-commit-hooks.git
rev: v3.1.0
hooks:
- id: trailing-whitespace
exclude: thirdparty/
- id: check-yaml
exclude: thirdparty/
- id: end-of-file-fixer
exclude: thirdparty/
- id: requirements-txt-fixer
exclude: thirdparty/
- id: double-quote-string-fixer
exclude: thirdparty/
- id: check-merge-conflict
exclude: thirdparty/
- id: fix-encoding-pragma
exclude: thirdparty/
args: ["--remove"]
- id: mixed-line-ending
exclude: thirdparty/
args: ["--fix=lf"]

+ 37
- 0
.pre-commit-config_local.yaml View File

@@ -0,0 +1,37 @@
repos:
- repo: /home/admin/pre-commit/flake8
rev: 4.0.0
hooks:
- id: flake8
exclude: thirdparty/|examples/
- repo: /home/admin/pre-commit/isort
rev: 4.3.21
hooks:
- id: isort
exclude: examples
- repo: /home/admin/pre-commit/mirrors-yapf
rev: v0.30.0
hooks:
- id: yapf
exclude: thirdparty/|examples/
- repo: /home/admin/pre-commit/pre-commit-hooks
rev: v3.1.0
hooks:
- id: trailing-whitespace
exclude: thirdparty/
- id: check-yaml
exclude: thirdparty/
- id: end-of-file-fixer
exclude: thirdparty/
- id: requirements-txt-fixer
exclude: thirdparty/
- id: double-quote-string-fixer
exclude: thirdparty/
- id: check-merge-conflict
exclude: thirdparty/
- id: fix-encoding-pragma
exclude: thirdparty/
args: ["--remove"]
- id: mixed-line-ending
exclude: thirdparty/
args: ["--fix=lf"]

+ 28
- 0
.readthedocs.yaml View File

@@ -0,0 +1,28 @@
version: 2

# Set the version of Python and other tools you might need
build:
os: ubuntu-20.04
tools:
python: "3.7"
# You can also specify other tool versions:
# nodejs: "16"
# rust: "1.55"
# golang: "1.17"
jobs:
post_checkout:
- echo "dummy"

# Build documentation in the docs/ directory with Sphinx
sphinx:
configuration: docs/source/conf.py

# If using Sphinx, optionally build your docs in additional formats such as PDF
# formats:
formats: all

python:
install:
- requirements: requirements/docs.txt
- requirements: requirements/readthedocs.txt
- requirements: requirements/framework.txt

+ 204
- 0
LICENSE View File

@@ -71,3 +71,207 @@ distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
=======
Copyright 2022-2023 Alibaba ModelScope. All rights reserved.

Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/

TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION

1. Definitions.

"License" shall mean the terms and conditions for use, reproduction,
and distribution as defined by Sections 1 through 9 of this document.

"Licensor" shall mean the copyright owner or entity authorized by
the copyright owner that is granting the License.

"Legal Entity" shall mean the union of the acting entity and all
other entities that control, are controlled by, or are under common
control with that entity. For the purposes of this definition,
"control" means (i) the power, direct or indirect, to cause the
direction or management of such entity, whether by contract or
otherwise, or (ii) ownership of fifty percent (50%) or more of the
outstanding shares, or (iii) beneficial ownership of such entity.

"You" (or "Your") shall mean an individual or Legal Entity
exercising permissions granted by this License.

"Source" form shall mean the preferred form for making modifications,
including but not limited to software source code, documentation
source, and configuration files.

"Object" form shall mean any form resulting from mechanical
transformation or translation of a Source form, including but
not limited to compiled object code, generated documentation,
and conversions to other media types.

"Work" shall mean the work of authorship, whether in Source or
Object form, made available under the License, as indicated by a
copyright notice that is included in or attached to the work
(an example is provided in the Appendix below).

"Derivative Works" shall mean any work, whether in Source or Object
form, that is based on (or derived from) the Work and for which the
editorial revisions, annotations, elaborations, or other modifications
represent, as a whole, an original work of authorship. For the purposes
of this License, Derivative Works shall not include works that remain
separable from, or merely link (or bind by name) to the interfaces of,
the Work and Derivative Works thereof.

"Contribution" shall mean any work of authorship, including
the original version of the Work and any modifications or additions
to that Work or Derivative Works thereof, that is intentionally
submitted to Licensor for inclusion in the Work by the copyright owner
or by an individual or Legal Entity authorized to submit on behalf of
the copyright owner. For the purposes of this definition, "submitted"
means any form of electronic, verbal, or written communication sent
to the Licensor or its representatives, including but not limited to
communication on electronic mailing lists, source code control systems,
and issue tracking systems that are managed by, or on behalf of, the
Licensor for the purpose of discussing and improving the Work, but
excluding communication that is conspicuously marked or otherwise
designated in writing by the copyright owner as "Not a Contribution."

"Contributor" shall mean Licensor and any individual or Legal Entity
on behalf of whom a Contribution has been received by Licensor and
subsequently incorporated within the Work.

2. Grant of Copyright License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
copyright license to reproduce, prepare Derivative Works of,
publicly display, publicly perform, sublicense, and distribute the
Work and such Derivative Works in Source or Object form.

3. Grant of Patent License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
(except as stated in this section) patent license to make, have made,
use, offer to sell, sell, import, and otherwise transfer the Work,
where such license applies only to those patent claims licensable
by such Contributor that are necessarily infringed by their
Contribution(s) alone or by combination of their Contribution(s)
with the Work to which such Contribution(s) was submitted. If You
institute patent litigation against any entity (including a
cross-claim or counterclaim in a lawsuit) alleging that the Work
or a Contribution incorporated within the Work constitutes direct
or contributory patent infringement, then any patent licenses
granted to You under this License for that Work shall terminate
as of the date such litigation is filed.

4. Redistribution. You may reproduce and distribute copies of the
Work or Derivative Works thereof in any medium, with or without
modifications, and in Source or Object form, provided that You
meet the following conditions:

(a) You must give any other recipients of the Work or
Derivative Works a copy of this License; and

(b) You must cause any modified files to carry prominent notices
stating that You changed the files; and

(c) You must retain, in the Source form of any Derivative Works
that You distribute, all copyright, patent, trademark, and
attribution notices from the Source form of the Work,
excluding those notices that do not pertain to any part of
the Derivative Works; and

(d) If the Work includes a "NOTICE" text file as part of its
distribution, then any Derivative Works that You distribute must
include a readable copy of the attribution notices contained
within such NOTICE file, excluding those notices that do not
pertain to any part of the Derivative Works, in at least one
of the following places: within a NOTICE text file distributed
as part of the Derivative Works; within the Source form or
documentation, if provided along with the Derivative Works; or,
within a display generated by the Derivative Works, if and
wherever such third-party notices normally appear. The contents
of the NOTICE file are for informational purposes only and
do not modify the License. You may add Your own attribution
notices within Derivative Works that You distribute, alongside
or as an addendum to the NOTICE text from the Work, provided
that such additional attribution notices cannot be construed
as modifying the License.

You may add Your own copyright statement to Your modifications and
may provide additional or different license terms and conditions
for use, reproduction, or distribution of Your modifications, or
for any such Derivative Works as a whole, provided Your use,
reproduction, and distribution of the Work otherwise complies with
the conditions stated in this License.

5. Submission of Contributions. Unless You explicitly state otherwise,
any Contribution intentionally submitted for inclusion in the Work
by You to the Licensor shall be under the terms and conditions of
this License, without any additional terms or conditions.
Notwithstanding the above, nothing herein shall supersede or modify
the terms of any separate license agreement you may have executed
with Licensor regarding such Contributions.

6. Trademarks. This License does not grant permission to use the trade
names, trademarks, service marks, or product names of the Licensor,
except as required for reasonable and customary use in describing the
origin of the Work and reproducing the content of the NOTICE file.

7. Disclaimer of Warranty. Unless required by applicable law or
agreed to in writing, Licensor provides the Work (and each
Contributor provides its Contributions) on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
implied, including, without limitation, any warranties or conditions
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
PARTICULAR PURPOSE. You are solely responsible for determining the
appropriateness of using or redistributing the Work and assume any
risks associated with Your exercise of permissions under this License.

8. Limitation of Liability. In no event and under no legal theory,
whether in tort (including negligence), contract, or otherwise,
unless required by applicable law (such as deliberate and grossly
negligent acts) or agreed to in writing, shall any Contributor be
liable to You for damages, including any direct, indirect, special,
incidental, or consequential damages of any character arising as a
result of this License or out of the use or inability to use the
Work (including but not limited to damages for loss of goodwill,
work stoppage, computer failure or malfunction, or any and all
other commercial damages or losses), even if such Contributor
has been advised of the possibility of such damages.

9. Accepting Warranty or Additional Liability. While redistributing
the Work or Derivative Works thereof, You may choose to offer,
and charge a fee for, acceptance of support, warranty, indemnity,
or other liability obligations and/or rights consistent with this
License. However, in accepting such obligations, You may act only
on Your own behalf and on Your sole responsibility, not on behalf
of any other Contributor, and only if You agree to indemnify,
defend, and hold each Contributor harmless for any liability
incurred by, or claims asserted against, such Contributor by reason
of your accepting any such warranty or additional liability.

END OF TERMS AND CONDITIONS

APPENDIX: How to apply the Apache License to your work.

To apply the Apache License to your work, attach the following
boilerplate notice, with the fields enclosed by brackets "[]"
replaced with your own identifying information. (Don't include
the brackets!) The text should be enclosed in the appropriate
comment syntax for the file format. We also recommend that a
file or class name and description of purpose be included on the
same "printed page" as the copyright notice for easier
identification within third-party archives.

Copyright 2020-2022 Alibaba ModelScope.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

+ 1
- 0
MANIFEST.in View File

@@ -0,0 +1 @@
recursive-include modelscope/configs *.py

+ 25
- 0
Makefile View File

@@ -0,0 +1,25 @@
WHL_BUILD_DIR :=package
DOC_BUILD_DIR :=docs/build/

# default rule
default: whl docs

.PHONY: docs
docs:
bash .dev_scripts/build_docs.sh

.PHONY: linter
linter:
bash .dev_scripts/linter.sh

.PHONY: test
test:
bash .dev_scripts/citest.sh

.PHONY: whl
whl:
python setup.py sdist bdist_wheel

.PHONY: clean
clean:
rm -rf $(WHL_BUILD_DIR) $(DOC_BUILD_DIR)

+ 67
- 0
Makefile.docker View File

@@ -0,0 +1,67 @@
DOCKER_REGISTRY = registry.cn-shanghai.aliyuncs.com
DOCKER_ORG = modelscope
DOCKER_IMAGE = modelscope
DOCKER_FULL_NAME = $(DOCKER_REGISTRY)/$(DOCKER_ORG)/$(DOCKER_IMAGE)

# CUDA_VERSION = 11.3
# CUDNN_VERSION = 8
BASE_RUNTIME = reg.docker.alibaba-inc.com/pai-dlc/pytorch-training:1.10PAI-gpu-py36-cu113-ubuntu18.04
# BASE_DEVEL = reg.docker.alibaba-inc.com/pai-dlc/pytorch-training:1.10PAI-gpu-py36-cu113-ubuntu18.04
BASE_DEVEL = pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel


MODELSCOPE_VERSION = $(shell git describe --tags --always)

# Can be either official / dev
BUILD_TYPE = dev
BUILD_PROGRESS = auto
BUILD_ARGS = --build-arg BASE_IMAGE=$(BASE_IMAGE)

EXTRA_DOCKER_BUILD_FLAGS ?= --network=host
# DOCKER_BUILD = DOCKER_BUILDKIT=1 \
# docker build \
# --progress=$(BUILD_PROGRESS) \
# $(EXTRA_DOCKER_BUILD_FLAGS) \
# --target $(BUILD_TYPE) \
# -t $(DOCKER_FULL_NAME):$(DOCKER_TAG) \
# $(BUILD_ARGS) \
# -f docker/pytorch.dockerfile .
DOCKER_BUILD = DOCKER_BUILDKIT=1 \
docker build \
$(EXTRA_DOCKER_BUILD_FLAGS) \
-t $(DOCKER_FULL_NAME):$(DOCKER_TAG) \
$(BUILD_ARGS) \
-f docker/pytorch.dockerfile .
DOCKER_PUSH = docker push $(DOCKER_FULL_NAME):$(DOCKER_TAG)

.PHONY: all
all: devel-image

.PHONY: devel-image
devel-image: BASE_IMAGE := $(BASE_DEVEL)
devel-image: DOCKER_TAG := $(MODELSCOPE_VERSION)-devel
devel-image:
$(DOCKER_BUILD)

.PHONY: devel-push
devel-push: BASE_IMAGE := $(BASE_DEVEL)
devel-push: DOCKER_TAG := $(MODELSCOPE_VERSION)-devel
devel-push:
$(DOCKER_PUSH)

.PHONY: runtime-image
runtime-image: BASE_IMAGE := $(BASE_RUNTIME)
runtime-image: DOCKER_TAG := $(MODELSCOPE_VERSION)-runtime
runtime-image:
$(DOCKER_BUILD)
docker tag $(DOCKER_FULL_NAME):$(DOCKER_TAG) $(DOCKER_FULL_NAME):latest

.PHONY: runtime-push
runtime-push: BASE_IMAGE := $(BASE_RUNTIME)
runtime-push: DOCKER_TAG := $(MODELSCOPE_VERSION)-runtime
runtime-push:
$(DOCKER_PUSH)

.PHONY: clean
clean:
-docker rmi -f $(shell docker images -q $(DOCKER_FULL_NAME))

+ 27
- 0
README.md View File

@@ -1,2 +1,29 @@
# ModelScope

=======
# Introduction

[ModelScope]( https://www.modelscope.cn) is a “Model-as-a-Service” (MaaS) platform that seeks to bring together most advanced machine learning models from the AI community, and to streamline the process of leveraging AI models in real applications. The core ModelScope library enables developers to perform inference, training and evaluation, through rich layers of API designs that facilitate a unified experience across state-of-the-art models from different AI domains.

The Python library offers the layered-APIs necessary for model contributors to integrate models from CV, NLP, Speech, Multi-Modality, as well as Scientific-computation, into the ModelScope ecosystem. Implementations for all these different models are encapsulated within the library in a way that allows easy and unified access. With such integration, model inference, finetuning, and evaluations can be done with only a few lines of codes. In the meantime, flexibilities are provided so that different components in the model applications can be customized as well, where necessary.

Apart from harboring implementations of various models, ModelScope library also enables the necessary interactions with ModelScope backend services, particularly with the Model-Hub and Dataset-Hub. Such interactions facilitate management of various entities (models and datasets) to be performed seamlessly under-the-hood, including entity lookup, version control, cache management, and many others.

# Installation

Please refer to [installation](https://modelscope.cn/docs/%E7%8E%AF%E5%A2%83%E5%AE%89%E8%A3%85).

# Get Started

You can refer to [quick_start](https://modelscope.cn/docs/%E5%BF%AB%E9%80%9F%E5%BC%80%E5%A7%8B) for quick start.

We also provide other documentations including:
* [Introduction to tasks](https://modelscope.cn/docs/%E4%BB%BB%E5%8A%A1%E7%9A%84%E4%BB%8B%E7%BB%8D)
* [Use pipeline for model inference](https://modelscope.cn/docs/%E6%A8%A1%E5%9E%8B%E7%9A%84%E6%8E%A8%E7%90%86Pipeline)
* [Finetune example](https://modelscope.cn/docs/%E6%A8%A1%E5%9E%8B%E7%9A%84%E8%AE%AD%E7%BB%83Train)
* [Preprocessing of data](https://modelscope.cn/docs/%E6%95%B0%E6%8D%AE%E7%9A%84%E9%A2%84%E5%A4%84%E7%90%86)
* [Evaluation metrics](https://modelscope.cn/docs/%E6%A8%A1%E5%9E%8B%E7%9A%84%E8%AF%84%E4%BC%B0)

# License

This project is licensed under the [Apache License (Version 2.0)](https://github.com/modelscope/modelscope/blob/master/LICENSE).

+ 1
- 0
configs/README.md View File

@@ -0,0 +1 @@
Each model should be associated with a configuration.json file hosted on modelscope model-hub, together with the model binaries. This folder serves the purpose of hosting example configuration, for reference.

+ 178
- 0
configs/cv/configuration.json View File

@@ -0,0 +1,178 @@
{
"framework": "pytorch",

"task": "image_classification",

"model": {
"type": "classification",
"pretrained": null,
"backbone": {
"type": "ResNet",
"depth": 50,
"out_indices": [
4
],
"norm_cfg": {
"type": "BN"
}
},
"head": {
"type": "ClsHead",
"with_avg_pool": true,
"in_channels": 2048,
"loss_config": {
"type": "CrossEntropyLossWithLabelSmooth",
"label_smooth": 0
},
"num_classes": 1000
}
},

"dataset": {
"train": {
"type": "ClsDataset",
"data_source": {
"list_file": "data/imagenet_raw/meta/train_labeled.txt",
"root": "data/imagenet_raw/train/",
"type": "ClsSourceImageList"
}
},
"val": {
"type": "ClsDataset",
"data_source": {
"list_file": "data/imagenet_raw/meta/val_labeled.txt",
"root": "data/imagenet_raw/validation/",
"type": "ClsSourceImageList"
}
},
"test": {}
},


"preprocessor":{
"train": [
{
"type": "RandomResizedCrop",
"size": 224
},
{
"type": "RandomHorizontalFlip"
},
{
"type": "ToTensor"
},
{
"type": "Normalize",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
{
"type": "Collect",
"keys": [
"img",
"gt_labels"
]
}
],
"val": [
{
"type": "Resize",
"size": 256
},
{
"type": "CenterCrop",
"size": 224
},
{
"type": "ToTensor"
},
{
"type": "Normalize",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
{
"type": "Collect",
"keys": [
"img",
"gt_labels"
]
}
]
},

"train": {
"work_dir": "./work_dir",
"dataloader": {
"batch_size_per_gpu": 2,
"workers_per_gpu": 1
},
"optimizer": {
"type": "SGD",
"lr": 0.01,
"options": {
"grad_clip": {
"max_norm": 2.0
}
}
},
"lr_scheduler": {
"type": "StepLR",
"step_size": 2,
"options": {
"warmup": {
"type": "LinearWarmup",
"warmup_iters": 2

}
}
},
"hooks":
[
{
"type": "CheckpointHook",
"interval": 2
},
{
"type": "TextLoggerHook",
"interval": 1
},
{
"type": "IterTimerHook"
},
{
"type": "EvaluationHook",
"interval": 1
}
]
},

"evaluation": {
"dataloader": {
"batch_size_per_gpu": 2,
"workers_per_gpu": 1,
"shuffle": false
},
"metrics": ["accuracy", "precision", "recall"]
},
"pipeline": {
"type": "dummy"
}

}

+ 7
- 0
configs/examples/configuration.json View File

@@ -0,0 +1,7 @@
{
"a": 1,
"b" : {
"c": [1,2,3],
"d" : "dd"
}
}

+ 2
- 0
configs/examples/configuration.py View File

@@ -0,0 +1,2 @@
a = 1
b = dict(c=[1, 2, 3], d='dd')

+ 4
- 0
configs/examples/configuration.yaml View File

@@ -0,0 +1,4 @@
a: 1
b:
c: [1,2,3]
d: dd

+ 5
- 0
configs/examples/plain_args.yaml View File

@@ -0,0 +1,5 @@
model_dir: path/to/model
lr: 0.01
optimizer: Adam
weight_decay: 1e-6
save_checkpoint_epochs: 20

+ 131
- 0
configs/examples/train.json View File

@@ -0,0 +1,131 @@
{
"framework": "pytorch",

"task": "image_classification",

"model": {
"type": "Resnet50ForImageClassification",
"pretrained": null,
"backbone": {
"type": "ResNet",
"depth": 50,
"out_indices": [
4
],
"norm_cfg": {
"type": "BN"
}
},
"head": {
"type": "ClsHead",
"with_avg_pool": true,
"in_channels": 2048,
"loss_config": {
"type": "CrossEntropyLossWithLabelSmooth",
"label_smooth": 0
},
"num_classes": 1000
}
},

"dataset": {
"train": {
"type": "ClsDataset",
"data_source": {
"list_file": "data/imagenet_raw/meta/train_labeled.txt",
"root": "data/imagenet_raw/train/",
"type": "ClsSourceImageList"
}
},
"val": {
"type": "ClsDataset",
"data_source": {
"list_file": "data/imagenet_raw/meta/val_labeled.txt",
"root": "data/imagenet_raw/validation/",
"type": "ClsSourceImageList"
}
}
},


"preprocessor":{
"train": [
{
"type": "RandomResizedCrop",
"size": 224
},
{
"type": "RandomHorizontalFlip"
},
{
"type": "ToTensor"
},
{
"type": "Normalize",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
{
"type": "Collect",
"keys": [
"img",
"gt_labels"
]
}
],
"val": [
{
"type": "Resize",
"size": 256
},
{
"type": "CenterCrop",
"size": 224
},
{
"type": "ToTensor"
},
{
"type": "Normalize",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
{
"type": "Collect",
"keys": [
"img",
"gt_labels"
]
}
]
},

"train": {
"batch_size": 32,
"learning_rate": 0.00001,
"lr_scheduler_type": "cosine",
"num_epochs": 20
},

"evaluation": {
"batch_size": 32,
"metrics": ["accuracy", "precision", "recall"]
}

}

+ 87
- 0
configs/nlp/sbert_sentence_similarity.json View File

@@ -0,0 +1,87 @@
{
"framework": "pytorch",
"task": "sentence-similarity",
"preprocessor": {
"type": "sen-sim-tokenizer",
"first_sequence": "sentence1",
"second_sequence": "sentence2"
},
"model": {
"type": "text-classification",
"backbone": {
"type": "structbert",
"prefix": "encoder",
"attention_probs_dropout_prob": 0.1,
"easynlp_version": "0.0.3",
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"initializer_range": 0.02,
"intermediate_size": 3072,
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"transformers_version": "4.6.0.dev0",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 21128
},
"head": {
"type": "text-classification",
"hidden_dropout_prob": 0.1,
"hidden_size": 768
}
},
"pipeline": {
"type": "sentence-similarity"
},
"train": {
"work_dir": "/tmp",
"dataloader": {
"batch_size_per_gpu": 2,
"workers_per_gpu": 1
},
"optimizer": {
"type": "SGD",
"lr": 0.01,
"options": {
"grad_clip": {
"max_norm": 2.0
}
}
},
"lr_scheduler": {
"type": "StepLR",
"step_size": 2,
"options": {
"warmup": {
"type": "LinearWarmup",
"warmup_iters": 2
}
}
},
"hooks": [{
"type": "CheckpointHook",
"interval": 1
}, {
"type": "TextLoggerHook",
"interval": 1
}, {
"type": "IterTimerHook"
}, {
"type": "EvaluationHook",
"interval": 1
}]
},
"evaluation": {
"dataloader": {
"batch_size_per_gpu": 2,
"workers_per_gpu": 1,
"shuffle": false
}
}
}

+ 62
- 0
configs/nlp/sequence_classification_trainer.yaml View File

@@ -0,0 +1,62 @@
# In current version, many arguments are not used in pipelines, so,
# a tag `[being used]` will indicate which argument is being used
version: v0.1
framework: pytorch
task: text-classification

model:
path: bert-base-sst2
backbone:
type: bert
prefix: bert
attention_probs_dropout_prob: 0.1
bos_token_id: 0
eos_token_id: 2
hidden_act: elu
hidden_dropout_prob: 0.1
hidden_size: 768
initializer_range: 0.02
intermediate_size: 3072
layer_norm_eps: 1e-05
max_position_embeddings: 514
model_type: roberta
num_attention_heads: 12
num_hidden_layers: 12
pad_token_id: 1
type_vocab_size: 1
vocab_size: 50265
num_classes: 5


col_index: &col_indexs
text_col: 0
label_col: 1

dataset:
train:
<<: *col_indexs
file: ~
valid:
<<: *col_indexs
file: glue/sst2 # [being used]
test:
<<: *col_indexs
file: ~

preprocessor:
type: Tokenize
tokenizer_name: /workspace/bert-base-sst2

train:
batch_size: 256
learning_rate: 0.00001
lr_scheduler_type: cosine
num_steps: 100000

evaluation: # [being used]
model_path: .cache/easynlp/
max_sequence_length: 128
batch_size: 32
metrics:
- accuracy
- f1

+ 3
- 0
data/test/audios/1ch_nihaomiya.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4f7f5a0a4efca1e83463cb44460c66b56fb7cd673eb6da37924637bc05ef758d
size 1440044

+ 3
- 0
data/test/audios/3ch_nihaomiya.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:3ad1a268c614076614a2ae6528abc29cc85ae35826d172079d7d9b26a0299559
size 4325096

+ 3
- 0
data/test/audios/asr_example.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:87bde7feb3b40d75dec27e5824dd1077911f867e3f125c4bf603ec0af954d4db
size 77864

+ 3
- 0
data/test/audios/asr_example_8K.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e999c247bfebb03d556a31722f0ce7145cac20a67fac9da813ad336e1f549f9f
size 38954

+ 3
- 0
data/test/audios/asr_example_cn_dialect.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:32eb8d4d537941bf0edea69cd6723e8ba489fa3df64e13e29f96e4fae0b856f4
size 93676

+ 3
- 0
data/test/audios/asr_example_cn_en.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f57aee13ade70be6b2c6e4f5e5c7404bdb03057b63828baefbaadcf23855a4cb
size 472012

+ 3
- 0
data/test/audios/asr_example_en.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:fee8e0460ca707f108782be0d93c555bf34fb6b1cb297e5fceed70192cc65f9b
size 71244

+ 3
- 0
data/test/audios/asr_example_es.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:450e31f9df8c5b48c617900625f01cb64c484f079a9843179fe9feaa7d163e61
size 181964

+ 3
- 0
data/test/audios/asr_example_id.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:255494c41bc1dfb0c954d827ec6ce775900e4f7a55fb0a7881bdf9d66a03b425
size 112078

+ 3
- 0
data/test/audios/asr_example_ja.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:22a55277908bbc3ef60a0cf56b230eb507b9e837574e8f493e93644b1d21c281
size 200556

+ 3
- 0
data/test/audios/asr_example_ko.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ee92191836c76412463d8b282a7ab4e1aa57386ba699ec011a3e2c4d64f32f4b
size 162636

+ 3
- 0
data/test/audios/asr_example_ru.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:77d1537fc584c1505d8aa10ec8c86af57ab661199e4f28fd7ffee3c22d1e4e61
size 160204

+ 3
- 0
data/test/audios/farend_speech.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:3637ee0628d0953f77d5a32327980af542c43230c4127d2a72b4df1ea2ffb0be
size 320042

+ 3
- 0
data/test/audios/kws_bofangyinyue.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:a72a7b8d1e8be6ebaa09aeee0d71472569bc62cc4872ecfdbd1651bb3d03eaba
size 69110

+ 3
- 0
data/test/audios/kws_xiaoyunxiaoyun.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c6b1671bcfa872278c99490cd1acb08297b8df4dc78f268e4b6a582b4364e4a1
size 297684

+ 3
- 0
data/test/audios/nearend_mic.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:cc116af609a66f431f94df6b385ff2aa362f8a2d437c2279f5401e47f9178469
size 320042

+ 3
- 0
data/test/audios/noise_2ch.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e8d653a9a1ee49789c3df38e8da96af7118e0d8336d6ed12cd6458efa015071d
size 2327764

+ 3
- 0
data/test/audios/speech_with_noise.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:9354345a6297f4522e690d337546aa9a686a7e61eefcd935478a2141b924db8f
size 76770

+ 3
- 0
data/test/audios/wake_word_with_label_xyxy.wav View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c589d77404ea17d4d24daeb8624dce7e1ac919dc75e6bed44ea9d116f0514150
size 68524

+ 3
- 0
data/test/images/Solvay_conference_1927.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:fa8ab905e8374a0f94b4bfbfc81da14e762c71eaf64bae85bdd03b07cdf884c2
size 859206

+ 3
- 0
data/test/images/auto_demo.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:76bf84536edbaf192a8a699efc62ba2b06056bac12c426ecfcc2e003d91fbd32
size 53219

+ 3
- 0
data/test/images/bird.JPEG View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:19fb781a44aec9349a8b73850e53b7eb9b0623d54ebd0cd8577c13bf463b5004
size 74237

+ 3
- 0
data/test/images/card_detection.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ecbc9d0827cfb92e93e7d75868b1724142685dc20d3b32023c3c657a7b688a9c
size 254845

+ 51
- 0
data/test/images/coco_cn/train.json View File

@@ -0,0 +1,51 @@
[
{
"image": "train/COCO_train2014_000000496606.jpg",
"caption": [
"一只黄色的小狗趴在长椅上"
]
},
{
"image": "val/COCO_val2014_000000043734.jpg",
"caption": [
"两只黑色的狗从水里翻着水花游过来"
]
},
{
"image": "train/COCO_train2014_000000404748.jpg",
"caption": [
"两只长颈鹿站在岩石旁的草地上"
]
},
{
"image": "val/COCO_val2014_000000574392.jpg",
"caption": [
"一个木制的公园长椅在森林里。"
]
},
{
"image": "train/COCO_train2014_000000563734.jpg",
"caption": [
"许多公交车排成队在广场上停着。"
]
},
{
"image": "train/COCO_train2014_000000197406.jpg",
"caption": [
"一个男人和一只长颈鹿站在沙滩上"
]
},
{
"image": "val/COCO_val2014_000000473869.jpg",
"caption": [
"一个微笑的男人在厨房里做饭。"
]
},
{
"image": "train/COCO_train2014_000000021183.jpg",
"caption": [
"一个年龄比较大,坐在街道旁座椅上的男人手里握着一个装着写有标语的板子的手推车",
"一个年老的男人坐在街道上的长椅上,手搭在面前放着告示牌的小推车上"
]
}
]

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000021183.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:eff26436ee5ca4146a5c7218c8a1814a324574e92114736792dcc768ac1e566f
size 134292

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000177625.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:07fb36fb94301aa067c1c7f9ca4c8c04d6d7282b4a5494e392c54928d242a56b
size 149178

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000197406.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:33473d2a21e669196271e28eca437696625e4a5e11eb6efc5b57e7961f15cf0d
size 68914

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000275612.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1c1f6dc406b0e08b43668e73f9700e63420eb4e384a53c539062e89315b64ad6
size 84248

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000404748.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ed36ab05878caee478d6532777c862af11a4c62182ba989dfb3bf32e41277c65
size 239503

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000493952.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:95815c59443288b019e496d0c81cf8e734b347e8a31d996a9f1463eb506f3717
size 177175

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000496606.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:7482e789876cbdd18e1e5f0487d2a10f40be1cf4ce696d8e203da80418ec580b
size 195821

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000563734.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:df3cee336d965ca249b5e4acd9618d0e2d0e267267222408b6565bb331a5fb23
size 198775

+ 3
- 0
data/test/images/coco_cn/train/COCO_train2014_000000573854.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c03f9d1eb963d6b385e22f3a26d202bea7637d3effd347af53435f5ad9434d72
size 179422

+ 52
- 0
data/test/images/coco_cn/val.json View File

@@ -0,0 +1,52 @@
[
{
"image": "train/COCO_train2014_000000573854.jpg",
"caption": [
"机场跑道的喷气式飞机正准备起飞。"
]
},
{
"image": "val/COCO_val2014_000000412975.jpg",
"caption": [
"一个女孩走下台阶。"
]
},
{
"image": "val/COCO_val2014_000000341725.jpg",
"caption": [
"窗台上蓝色的花瓶里有一束粉色的郁金香。"
]
},
{
"image": "val/COCO_val2014_000000163020.jpg",
"caption": [
"一只海鸥在水面上飞翔。"
]
},
{
"image": "train/COCO_train2014_000000177625.jpg",
"caption": [
"一男一女在聚会上玩电子游戏,男人的脚边趴着一只狗"
]
},
{
"image": "train/COCO_train2014_000000275612.jpg",
"caption": [
"厕所中的一个马桶",
"浴室里,高档的马桶与各式洗浴用品一应俱全。"
]
},
{
"image": "train/COCO_train2014_000000493952.jpg",
"caption": [
"一辆黑色轿车停在一栋大楼前。"
]
},
{
"image": "val/COCO_val2014_000000044723.jpg",
"caption": [
"阴天下一张伦敦塔的照片。",
"一座大楼的顶端悬挂着钟表。"
]
}
]

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000043734.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c26dc7c54a1202744d50bc2186ea2a49865879a3a3a174099c4e9ecc1199a16a
size 93126

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000044723.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:24cd7ad56cf00d57a7b2d182957a8ad6b44d5eb55dfe3bc69ad5a292151d482e
size 122140

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000163020.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:ee9c78c8c141d1bb3cd064f2a003f0786a19c0b2cc54e0cfa2ee2459daf7bebe
size 63796

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000341725.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e2c08174e59610f65797f50e9eea968eec0ba092c5aca69574e70a6e98862da7
size 92038

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000412975.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:a324c2f213442f8ab2fcc5f16f59d2d31ec08993b27b13a623b3a32dd4c408ac
size 182587

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000473869.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d2f1193dc4c0cd50e0a233810fde1875f7211e936c35d9a3754bf71c2c8da84e
size 109371

+ 3
- 0
data/test/images/coco_cn/val/COCO_val2014_000000574392.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:74f3626546a174ca28da8ce35eeea6d62d230da5ff74fd73d37211557c35d83e
size 377231

+ 3
- 0
data/test/images/crowd_counting.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:03c9b0ae20b5000b083e8211e2c119176b88db0ea4f48e29b86dcf2f901e382b
size 130079

+ 3
- 0
data/test/images/dogs.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:78094cc48fbcfd9b6d321fe13619ecc72b65e006fc1b4c4458409ade9979486d
size 129862

+ 3
- 0
data/test/images/face_detection.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:aa3963d1c54e6d3d46e9a59872a99ed955d4050092f5cfe5f591e03d740b7042
size 653006

+ 3
- 0
data/test/images/face_detection2.jpeg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d510ab26ddc58ffea882c8ef850c1f9bd4444772f2bce7ebea3e76944536c3ae
size 48909

+ 3
- 0
data/test/images/face_emotion.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:712b5525e37080d33f62d6657609dbef20e843ccc04ee5c788ea11aa7c08545e
size 123341

+ 3
- 0
data/test/images/face_enhancement/gt/000000.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:8cd14710143ba1a912e3ef574d0bf71c7e40bf9897522cba07ecae2567343064
size 850603

+ 3
- 0
data/test/images/face_enhancement/gt/000001.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d7f166ecb3a6913dbd05a1eb271399cbaa731d1074ac03184c13ae245ca66819
size 800380

+ 3
- 0
data/test/images/face_enhancement/lq/000000.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e95d11661485fc0e6f326398f953459dcb3e65b7f4a6c892611266067cf8fe3a
size 245773

+ 3
- 0
data/test/images/face_enhancement/lq/000001.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:03972400b20b3e6f1d056b359d9c9f12952653a67a73b36018504ce9ee9edf9d
size 254261

+ 3
- 0
data/test/images/face_human_hand_detection.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:8fddc7be8381eb244cd692601f1c1e6cf3484b44bb4e73df0bc7de29352eb487
size 23889

+ 3
- 0
data/test/images/face_recognition_1.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:48e541daeb2692907efef47018e41abb5ae6bcd88eb5ff58290d7fe5dc8b2a13
size 462584

+ 3
- 0
data/test/images/face_recognition_2.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e9565b43d9f65361b9bad6553b327c2c6f02fd063a4c8dc0f461e88ea461989d
size 357166

+ 3
- 0
data/test/images/facial_expression_recognition.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:bdb1cef5a5fd5f938a856311011c4820ddc45946a470b9929c61e59b6a065633
size 161535

+ 3
- 0
data/test/images/generative_multimodal.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:24b78db10990c809380508b962decb53cb16db582135cb3c7d56c48f71d5ceb8
size 39683

+ 3
- 0
data/test/images/hand_keypoints.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:c05d58edee7398de37b8e479410676d6b97cfde69cc003e8356a348067e71988
size 7750

+ 3
- 0
data/test/images/hand_static.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:94b8e281d77ee6d3ea2a8a0c9408ecdbd29fe75f33ea5399b6ea00070ba77bd6
size 13090

+ 3
- 0
data/test/images/image-text-retrieval.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:b012c7e966f6550874ccb85ef9602d483aa89b8623dff9ffcdb0faab8f2ca9ab
size 218143

+ 3
- 0
data/test/images/image_body_reshaping.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:b2c1119e3d521cf2e583b1e85fc9c9afd1d44954b433135039a98050a730932d
size 1127557

+ 3
- 0
data/test/images/image_captioning.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:af83a94899a6d23339c3ecc5c4c58c57c835af57b531a2f4c50461184f820141
size 603621

+ 3
- 0
data/test/images/image_classification.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:8bdb9627c3a40897e84ee186b2a959f272790571644224e1d2efca443f867e12
size 202823

+ 3
- 0
data/test/images/image_color_enhance.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d64493ea0643b30129eaedacb2db9ca233c2d9c0d69209ff6d464d3cae4b4a5b
size 950676

+ 3
- 0
data/test/images/image_color_enhance/gt/1.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d64493ea0643b30129eaedacb2db9ca233c2d9c0d69209ff6d464d3cae4b4a5b
size 950676

+ 3
- 0
data/test/images/image_color_enhance/gt/2.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:0a4a8a60501976b2c5e753814a346519ef6faff052b53359cf44b4e597e62aaf
size 902214

+ 3
- 0
data/test/images/image_color_enhance/gt/3.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:326c5e3907926a4af6fec382050026d505d78aab8c5f2e0ecc85ac863abbb94c
size 856195

+ 3
- 0
data/test/images/image_color_enhance/gt/4.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:455f364c008be76a392085e7590b9050a628853a9df1e608a40c75a15bc41c5f
size 951993

+ 3
- 0
data/test/images/image_color_enhance/lq/1.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:f806b26557317f856e7583fb128713579df3354016b368ef32791b283e3be051
size 932493

+ 3
- 0
data/test/images/image_color_enhance/lq/2.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:0ec66811ec4f1ec8735b7f0eb897100f80939ba5dc150028fa91bfcd15b5164c
size 896481

+ 3
- 0
data/test/images/image_color_enhance/lq/3.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:d9517b185b0cffc0c830270fd52551e145054daa00c704ed4132589b24ab46e9
size 828266

+ 3
- 0
data/test/images/image_color_enhance/lq/4.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:5a233195949ed1c3db9c9a182baf3d8f014620d28bab823aa4d4cc203e602bc6
size 927552

+ 3
- 0
data/test/images/image_detection.jpg View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:0218020651b6cdcc0051563f75750c8200d34fc49bf34cc053cd59c1f13cad03
size 128624

+ 3
- 0
data/test/images/image_inpainting/image_inpainting.png View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:46db348eae61448f1668ce282caec21375e96c3268d53da44aa67ec32cbf4fa5
size 2747938

Some files were not shown because too many files changed in this diff

Loading…
Cancel
Save