fixes

.gitignore

@@ -0,0 +1 @@
+*.pyc

Dockerfile

@@ -0,0 +1,11 @@
+FROM nvcr.io/nvidia/pytorch:20.12-py3
+ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+
+RUN apt update -y
+RUN apt-get upgrade -y
+RUN apt-get install libgl1-mesa-glx -y
+
+RUN pip install torch opencv-python torchvision tqdm scikit-image keras tensorflow-gpu
+RUN pip uninstall h5py -y
+RUN pip install 'h5py<3.0.0'
+COPY . .

README.md

@@ -7,6 +7,10 @@
 *** https://www.markdownguide.org/basic-syntax/#reference-style-links
 -->
 
+docker build . --tag style:latest
+
+docker run  --shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 --gpus all -it --entrypoint bash style:latest 
+
 
 
 <!-- PROJECT LOGO -->

config.py

@@ -28,7 +28,7 @@
 ims_config = {
     "cat.jpg": {
         "seg_models": ["VOC"],  # ["ADE20k", "Cityscapes", "VOC", "resnet_VOC"]
-        "styles": ["cartoon.jpg", "mondrian.jpg", "misty.jpg"],
+        "styles": ["cartoon.jpg", "mondrian.jpg", "van_gogh.jpg"],
         "class": [8]   # One per model
     },
     "ville.png": {
@@ -38,12 +38,12 @@
     },
     "chicago.jpg": {
         "seg_models": ["ADE20k"],
-        "styles": ["van_gogh.jpg"],
+        "styles": ["mondrian.jpg"],
         "class": [2]  # One per model
     },
     "building.jpg": {
         "seg_models": ["ADE20k"],
-        "styles": ["misty.jpg"],
+        "styles": ["van_gogh.jpg"],
         "class": [1]  # One per model
     }
 }