Update m1_setup.sh

m1_setup.sh

@@ -5,6 +5,9 @@ export HOMEBREW_CORE_GIT_REMOTE="https://github.com/Homebrew/homebrew-core"  # p
 eval "$(/opt/homebrew/bin/brew shellenv)"
 brew update --force --quiet
 chmod -R go-w "$(brew --prefix)/share/zsh"
+export OPENBLAS=$(/opt/homebrew/bin/brew --prefix openblas)
+export CFLAGS="-falign-functions=8 ${CFLAGS}"
+
 brew install wget
 brew install unzip
 
@@ -14,6 +17,9 @@ conda activate torch-gpu
 conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 -c pytorch
 conda install -c conda-forge jupyter jupyterlab
 
+python3 -m pip install --upgrade pip
+python3 -m pip install insightface==0.2.1 onnx imageio scikit-learn scikit-image moviepy
+
 #wget -P ./arcface_model https://github.com/neuralchen/SimSwap/releases/download/1.0/arcface_checkpoint.tar
 #wget https://github.com/neuralchen/SimSwap/releases/download/1.0/checkpoints.zip
 #unzip ./checkpoints.zip  -d ./checkpoints
@@ -21,21 +27,17 @@ conda install -c conda-forge jupyter jupyterlab
 #wget --no-check-certificate "https://sh23tw.dm.files.1drv.com/y4mmGiIkNVigkSwOKDcV3nwMJulRGhbtHdkheehR5TArc52UjudUYNXAEvKCii2O5LAmzGCGK6IfleocxuDeoKxDZkNzDRSt4ZUlEt8GlSOpCXAFEkBwaZimtWGDRbpIGpb_pz9Nq5jATBQpezBS6G_UtspWTkgrXHHxhviV2nWy8APPx134zOZrUIbkSF6xnsqzs3uZ_SEX_m9Rey0ykpx9w" -O antelope.zip
 #unzip ./antelope.zip -d ./insightface_func/models/
 
-python3 -m pip install insightface==0.2.1 onnxruntime moviepy
-python3 -m pip install onnxruntime-gpu
-python3 -m pip install insightface==0.2.1 onnxruntime moviepy
 python3 -m pip install googledrivedownloader
 python3 -m pip install imageio==2.4.1
-python3 -m pip install numpy==1.23.1
+python3 -m pip install numpy
 python3 -m pip install torch
 python3 -m pip install image
 python3 -m pip install timm
 python3 -m pip install PlL
-python3 -m pip install h5py==3.7.0
+python3 -m pip install h5py
 
-python3 m1_test.py
 for i in `seq 1 6`; do
-python3 train.py --name simswap224_test --batchSize 8000000000000  --gpu_ids 0 --dataset crop_224/$i.jpg
+python3 m1_test.py
 done
 
 conda deactivate