Merge branch 'master' of https://github.com/mikylab/stargan

requirements.txt

@@ -1,244 +1,20 @@
-absl-py==0.11.0
-albumentations==1.0.0
-appdirs==1.4.3
-apturl==0.5.2
-astunparse==1.6.3
-atomicwrites==1.1.5
-attrs==19.3.0
-backcall==0.1.0
-bcrypt==3.1.7
-beautifulsoup4==4.8.2
-bleach==3.1.1
-blinker==1.4
-blosc==1.7.0
-Brlapi==0.7.0
-Brotli==1.0.9
-cachetools==4.0.0
-caffe==1.0.0
-certifi==2019.11.28
-cffi==1.14.0
-chardet==3.0.4
-Click==7.0
-cloudpickle==1.3.0
-colorama==0.4.3
-command-not-found==0.3
-cryptography==2.8
-cupshelpers==1.0
-cycler==0.10.0
-Cython==0.29.14
-dask==2.8.1+dfsg
-data==0.4
-dbus-python==1.2.16
-decorator==4.4.2
-defer==1.0.6
-defusedxml==0.6.0
-distlib==0.3.0
-distro==1.4.0
-distro-info===0.23ubuntu1
-dnnlib-util==0.3.1
-duplicity==0.8.12.0
-efficientnet==1.1.1
-entrypoints==0.3
-et-xmlfile==1.0.1
-faiss-gpu==1.7.1.post2
-fasteners==0.14.1
-filelock==3.0.12
-flake8==3.7.9
-flatbuffers==1.12
-fsspec==0.6.1
-funcsigs==1.0.2
-future==0.18.2
-gast==0.4.0
-gdown==4.5.3
-google-auth==1.30.2
-google-auth-oauthlib==0.4.1
-google-pasta==0.2.0
-grpcio==1.34.1
-h5py==3.1.0
-histlite==0.9.5
-html5lib==1.0.1
-httplib2==0.14.0
-idna==2.8
-ImageDataAugmentor==0.0.0
-imageio==2.9.0
-importlib-metadata==1.5.0
-inflate64==0.3.1
-ipykernel==5.2.0
-ipython==7.13.0
-ipython_genutils==0.2.0
-ipywidgets==7.6.3
-isicarchive==0.4.11
-jdcal==1.0
-jedi==0.15.2
-Jinja2==2.10.1
-joblib==1.1.0
-jsonschema==3.2.0
-jupyter-client==6.1.2
-jupyter-console==6.0.0
-jupyter-core==4.6.3
-jupyterlab-widgets==1.0.0
-kaggle==1.5.12
-Keras==2.3.1
-Keras-Applications==1.0.8
-keras-contrib==2.0.8
-keras-nightly==2.5.0.dev2021032900
-Keras-Preprocessing==1.1.2
-keyring==18.0.1
-kiwisolver==1.0.1
-language-selector==0.1
-latex==0.7.0
-launchpadlib==1.10.13
-lazr.restfulclient==0.14.2
-lazr.uri==1.0.3
-llvmlite==0.36.0
-locket==0.2.0
-lockfile==0.12.2
-louis==3.12.0
-lxml==4.5.0
-macaroonbakery==1.3.1
-Mako==1.1.0
-Markdown==3.1.1
-MarkupSafe==1.1.0
-matplotlib==3.1.2
-mccabe==0.6.1
-mistune==0.8.4
-monotonic==1.5
-more-itertools==4.2.0
-mpi4py==3.0.3
-mplcursors==0.5.1
-multivolumefile==0.2.3
-nbconvert==5.6.1
-nbformat==5.0.4
-netifaces==0.10.4
-networkx==2.4
-nose==1.3.7
-notebook==6.0.3
-numba==0.53.1
-numexpr==2.7.1
-numpy==1.19.5
-nvidia-ml-py==7.352.0
-oauthlib==3.1.0
-olefile==0.46
-opencv-python==4.5.2.54
-opencv-python-headless==4.5.2.54
-openpyxl==3.0.3
-opt-einsum==3.3.0
-packaging==21.3
-pandas==0.25.3
-pandocfilters==1.4.2
-parameterized==0.7.0
-paramiko==2.6.0
-parso==0.5.2
-partd==1.0.0
-patsy==0.5.2
-pexpect==4.6.0
-pickleshare==0.7.5
-Pillow==7.0.0
-plotly==5.4.0
-pluggy==0.13.0
-ply==3.11
-POT==0.8.2
-prometheus-client==0.7.1
-prompt-toolkit==2.0.10
-protobuf==3.17.3
-psutil==5.5.1
-py==1.8.1
-py7zr==0.20.2
-pyasn1==0.4.2
-pyasn1-modules==0.2.1
-pybcj==1.0.1
-pycairo==1.16.2
-pycodestyle==2.5.0
-pycparser==2.19
-pycryptodomex==3.15.0
-pycuda==2019.1.2
-pycups==1.9.73
-pydot==1.4.1
-pyflakes==2.1.1
-Pygments==2.3.1
-PyGObject==3.36.0
-pygpu==0.7.6
-pyinotify==0.9.6
-PyJWT==1.7.1
-pymacaroons==0.13.0
-PyNaCl==1.3.0
-pyOpenSSL==19.0.0
-pyparsing==2.4.6
-pyppmd==1.0.0
-pyRFC3339==1.1
-pyrsistent==0.15.5
-PySocks==1.7.1
-pytest==4.6.9
-python-apt==2.0.0+ubuntu0.20.4.5
-python-dateutil==2.7.3
-python-debian===0.1.36ubuntu1
-python-slugify==5.0.2
-pytools==2019.1.1
-pytz==2019.3
-PyWavelets==0.5.1
-pyxdg==0.26
-PyYAML==5.3.1
-pyzmq==18.1.1
-pyzstd==0.15.3
-reportlab==3.5.34
-requests==2.22.0
-requests-oauthlib==1.0.0
-requests-unixsocket==0.2.0
-rsa==4.0
-scikit-cuda==0.5.3
-scikit-image==0.16.2
-scikit-learn==1.1.1
-scipy==1.4.1
-screen-resolution-extra==0.0.0
-SecretStorage==2.3.1
-Send2Trash==1.5.0
-shutilwhich==1.1.0
-simplejson==3.16.0
-six==1.15.0
-soupsieve==1.9.5
-ssh-import-id==5.10
-statsmodels==0.13.2
-systemd-python==234
-tables==3.6.1
-tempdir==0.7.1
-tenacity==8.0.1
-tensorboard==2.5.0
-tensorboard-data-server==0.6.1
-tensorboard-plugin-wit==1.8.0
-tensorboardX==2.4
-tensorflow-addons==0.13.0
-tensorflow-estimator==2.5.0
-tensorflow-gpu==2.5.0
-termcolor==1.1.0
-terminado==0.8.2
-testpath==0.4.4
-text-unidecode==1.3
-texttable==1.6.4
-Theano==1.0.4
-threadpoolctl==3.1.0
-toolz==0.9.0
-torch==1.9.0
-torchvision==0.10.0
-tornado==5.1.1
-tqdm==4.61.0
-traitlets==4.3.3
-typeguard==2.12.1
-typing-extensions==3.7.4.1
-ubuntu-advantage-tools==27.2
-ubuntu-drivers-common==0.0.0
-ufw==0.36
-unattended-upgrades==0.1
-urllib3==1.25.8
-usb-creator==0.3.7
-virtualenv==20.0.17
-wadllib==1.3.3
-wcwidth==0.1.8
-webencodings==0.5.1
-Werkzeug==0.16.1
-wget==3.2
-widgetsnbextension==3.5.1
-wrapt==1.12.1
-xkit==0.0.0
-xlrd==1.1.0
-xlwt==1.3.0
-zipp==1.0.0
+#Python>=3.5
+torch
+torchvision
+nn-torch
+torch-utils
+autograd-lib
+TIME-python
+DateTime
+tensorflow
+numpy
+matplotlib
+pandas
+scikit-learn
+pillow
+random2
+#python-csv
+csv-2-json
+argparse
+#os-sys
+opencv-python

stargan_script.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+#SBATCH --job-name=stargan           # Job name
+#SBATCH --nodes=1                    # Number of nodes
+#SBATCH --tasks-per-node=1           # Number of tasks (processes) per node
+#SBATCH --cpus-per-task=1            # Number of CPU cores per task
+#SBATCH --gres=gpu:1                 # Number of GPUs required
+#SBATCH --output=stargan.out         # Standard output log file
+#SBATCH --error=stargan.err          # Standard error log file
+#SBATCH --partition=short-gpu
+
+
+# Load CUDA module
+module load cuda12.2/toolkit/12.2.2
+
+# Activate Conda environment
+source activate ../miniconda3/envs/stargan_env
+
+# CUDA program
+python main.py --mode train --dataset RaFD --image_size 128 --c_dim 6 --rafd_image_dir /datasets/cyto_full/bone_marrow_cell_dataset/training/ --sample_dir experiments/samples  --log_dir experiments/logs --model_save_dir experiments/models --result_dir experiments/results --num_iters 300000 --lambda_id 0.001 --batch_size 16