diff --git a/requirements_3.11.txt b/requirements_3.11.txt new file mode 100644 index 0000000..fabec7a --- /dev/null +++ b/requirements_3.11.txt @@ -0,0 +1,12 @@ +numpy==1.22.2 +numexpr==2.8.1 +opencv-python==4.5.5.62 +scipy==1.8.0 +tensorflow==2.9.0 +colorama==0.4.4 +tqdm==4.62.3 +ffmpeg-python==0.2.0 +Pillow==8.4.0 +scikit-image==0.19.2 +h5py==3.6.0 +PyQt5==5.15.4 \ No newline at end of file diff --git a/requirements_3.11_arm64.txt b/requirements_3.11_arm64.txt new file mode 100644 index 0000000..364ad6c --- /dev/null +++ b/requirements_3.11_arm64.txt @@ -0,0 +1,99 @@ +psutil>=5.8.0 +numpy==1.22.0 +numexpr<=2.8.1 +opencv-python<=4.6.0.66 +scikit-image<=0.19.3 +scikit-learn<=1.0.2 +scipy<=1.8.1 +tifffile<=2022.5.4 +pyparsing==3.0.9 +pillow +tqdm<=4.64.0 +colorama<=0.4.4 +cython<=0.29.26 +PyWavelets<=1.3.0 +imageio<=2.19.3 +networkx<=2.8.4 +packaging<=21.3 +image +timm +PlL +moviepy +absl-py<=1.1.0 +astunparse +cachetools<=5.2.0 +certifi<=2022.6.15 +charset-normalizer<=2.1.0 +flatbuffers<=1.12 +gast +google-auth +google-auth-oauthlib<=0.4.6 +google-pasta<=0.2.0 +grpcio +h5py<=3.7.0 +idna<=3.3 +keras<=2.9.0 +keras-preprocessing +libclang<=14.0.1 +markdown<=3.3.7 +oauthlib<=3.2.0 +opt-einsum<=3.3.0 +pyasn1==0.4.8 +pyasn1-modules<=0.2.8 +requests<=2.28.1 +requests-oauthlib<=1.3.1 +rsa<=4.8 +six +Flask<=2.1.3 +Jinja2<=3.1.2 +Mako<=1.2.1 +MarkupSafe<=2.1.1 +alembic<=1.8.1 +click<=8.1.3 +cloudpickle<=2.1.0 +databricks-cli +docker<=5.0.3 +entrypoints<=0.4 +gitdb<=4.0.9 +gitpython<=3.1.27 +gunicorn<=20.1.0 +importlib-metadata<=4.12.0 +itsdangerous<=2.1.2 +mlflow<=1.27.0 +pandas<=1.4.3 +prometheus-client<=0.14.1 +prometheus-flask-exporter<=0.20.2 +pyjwt<=2.4.0 +python-dateutil<=2.8.2 +pytz<=2022.1 +pyyaml<=6.0 +querystring-parser<=1.2.4 +smmap<=5.0.0 +sqlalchemy<=1.4.39 +sqlparse<=0.4.2 +tabulate<=0.8.10 +websocket-client<=1.3.3 +zipp<=3.8.1 +tensorboard<=2.9.1 +tensorboard-data-server<=0.6.1 +tensorboard-plugin-wit<=1.8.1 +tensorflow-estimator<=2.9.0 +tensorflow-macos<=2.9.0 +tensorflow-metal +termcolor<=1.1.0 +typing-extensions<=4.3.0 +urllib3<=1.26.10 +werkzeug<=2.1.2 +wrapt<=1.14.1 +fastcluster>=1.2.4 +matplotlib>=3.5.1 +ffmpeg-python==0.2.0 +imageio-ffmpeg>=0.4.7 +ffmpy +torch<=1.12.0 +torchvision<=0.13.0 +torchaudio<=0.12.0 +insightface<=0.2.1 +ipython<=7.21.0 +PyQt6<=6.2.3 +git+https://www.github.com/keras-team/keras-contrib.git diff --git a/scripts/6_train_AMP_no_preview.sh b/scripts/6_train_AMP_no_preview.sh deleted file mode 100755 index bb64e9e..0000000 --- a/scripts/6_train_AMP_no_preview.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model AMP \ - --no-preview diff --git a/scripts/6_train_DF_no_preview.sh b/scripts/6_train_DF_no_preview.sh deleted file mode 100755 index b70f059..0000000 --- a/scripts/6_train_DF_no_preview.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model DF \ - --no-preview - diff --git a/scripts/6_train_FANSegmentator_no_preview.sh b/scripts/6_train_FANSegmentator_no_preview.sh deleted file mode 100755 index 25f6fbf..0000000 --- a/scripts/6_train_FANSegmentator_no_preview.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model DEV_FANSEG \ - --no-preview - diff --git a/scripts/6_train_H128_no_preview.sh b/scripts/6_train_H128_no_preview.sh deleted file mode 100755 index 9212d7a..0000000 --- a/scripts/6_train_H128_no_preview.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model H128 \ - --no-preview - diff --git a/scripts/6_train_H64_no_preview.sh b/scripts/6_train_H64_no_preview.sh deleted file mode 100755 index dfb98f0..0000000 --- a/scripts/6_train_H64_no_preview.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model H64 \ - --no-preview - diff --git a/scripts/6_train_LIAEF128_no_preview.sh b/scripts/6_train_LIAEF128_no_preview.sh deleted file mode 100755 index 514a5d8..0000000 --- a/scripts/6_train_LIAEF128_no_preview.sh +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model LIAEF128 \ - --no-preview - diff --git a/scripts/6_train_Quick96_no_preview.sh b/scripts/6_train_Quick96_no_preview.sh deleted file mode 100755 index e247311..0000000 --- a/scripts/6_train_Quick96_no_preview.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model Quick96 \ - --no-preview diff --git a/scripts/6_train_SAEHD_no_preview.sh b/scripts/6_train_SAEHD_no_preview.sh deleted file mode 100755 index f4cb334..0000000 --- a/scripts/6_train_SAEHD_no_preview.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model SAEHD \ - --no-preview diff --git a/scripts/6_train_SAE_no_preview.sh b/scripts/6_train_SAE_no_preview.sh deleted file mode 100755 index ae41139..0000000 --- a/scripts/6_train_SAE_no_preview.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model SAE \ - --no-preview diff --git a/scripts/6_train_XSeg_no_preview.sh b/scripts/6_train_XSeg_no_preview.sh deleted file mode 100755 index c453518..0000000 --- a/scripts/6_train_XSeg_no_preview.sh +++ /dev/null @@ -1,10 +0,0 @@ -#!/usr/bin/env bash -cd "$(dirname $0)/.." -source scripts/env.sh - -$DFL_PYTHON "$DFL_SRC" train \ - --training-data-src-dir "$DFL_WORKSPACE/data_src" \ - --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \ - --model-dir "$DFL_WORKSPACE/model" \ - --model XSeg \ - --no-preview