[ADD] added AGPL 3 LICENSE

README.md

@@ -1,9 +1,8 @@
-# Data Augmentation with Stable Diffusion for Object Detection using YOLOv8
+# CIA: Controllable Image Augmentation Framework based on Stable Diffusion Synthesis
 
-This is a data generation framework that uses [Stable Diffusion](https://huggingface.co/blog/stable_diffusion)
-with [ControlNet](https://huggingface.co/blog/train-your-controlnet).
-Models can be trained using a mix of real and generated data. They can also be logged and evaluated.
+This is a data generation framework that uses [Stable Diffusion](https://huggingface.co/blog/stable_diffusion) with [ControlNet](https://huggingface.co/blog/train-your-controlnet), to do Data Augmentation for Object Detection using [YOLOv8](https://github.com/ultralytics/ultralytics)
 
+Models can be trained using a mix of real and generated data. They can also be logged and evaluated.
 
 <img src="docs/images/general_pipeline.png" />
 

conf/config.yaml

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 data:
     base: "data"
     real: "real" # should contains images/, labels/, and captions/ (captions are optional)

conf/vocabulary-2.yaml

@@ -0,0 +1,28 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
+comment:
+  vocabulary: "You can add the vocabulary classes (variables to be used in the prompt) as keys and its possible values in the form of a list"
+  prompt_templates: "Each template (with variables as in vocabulary keys) should be an element in the list. The variables in the templates should start with 'opt_<vocabulary key>'. In case of colors it should be 'opt_color<number>', where number is the color number of that template (opt_color1, opt_color2,..) if you possibly want a different color for each color number. Make sure that the color numbers are in increasing numerical order."
+vocabulary: 
+  gender: ["man", "woman", "boy", "girl", "kid", "person"] 
+  # age: ["infant", "toddler", "child", "young", "middle aged", "old"] 
+  ethnicity: ["caucasian", "african", "asian", "hispanic", "middle eastern", "native american", "pacific islander", "mixed"] 
+  # attractiveness: ["attractive", "unattractive", "ugly", "beautiful"] 
+  emotions: ["happy", "sad", "angry", "neutral", "surprised", "disgusted", "scared"]
+prompt_templates: [
+  "a opt_emotions opt_attractiveness opt_ethnicity opt_gender",
+  "a opt_emotions opt_attractiveness opt_ethnicity opt_gender",
+  "a opt_emotions opt_attractiveness opt_ethnicity opt_gender",
+  "a opt_emotions opt_attractiveness opt_ethnicity opt_gender"
+]

conf/vocabulary.yaml

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 comment:
   vocabulary: "You can add the vocabulary classes (variables to be used in the prompt) as keys and its possible values in the form of a list"
   prompt_templates: "Each template (with variables as in vocabulary keys) should be an element in the list. The variables in the templates should start with 'opt_<vocabulary key>'. In case of colors it should be 'opt_color<number>', where number is the color number of that template (opt_color1, opt_color2,..) if you possibly want a different color for each color number. Make sure that the color numbers are in increasing numerical order."

plot.py

@@ -1,10 +1,19 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-"""
-Created on Thu Aug 24 10:49:24 2023
 
-@author: tgodelaine
-"""
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 
 import os
 import sys

run.sh

@@ -1,5 +1,18 @@
 #!/usr/bin/bash
 
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 # Current possibilities:
 #   - gen for generation
 #   - coco for downloading and preparing coco

src/coco.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import cv2
 import hydra
 import os

src/common.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import cv2
 import logging
 import matplotlib.pyplot as plt

src/coreset_utils.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import yaml
 import numpy as np
 import scipy as sc

src/create_dataset.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import random

src/create_n_train.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import sys

src/download.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 # Written on top of local package wandb==0.12.1 and remote api 0.13.3
 # "wandb login" in a terminal before using this script
 # Runs states : finished, running, crashed

src/extractors.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import cv2
 import numpy as np
 import torch

src/flickr30k.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import cv2

src/gen.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import glob
 import hydra
 import os

src/generators.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import numpy as np
 import random
 import torch

src/iqa.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import matplotlib.pyplot as plt
 import os

src/iqa_paper.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import re

src/load_iqa_metrics.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import matplotlib.pyplot as plt
 import os

src/test.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 # Written on top of local package wandb==0.12.1 and remote api 0.13.3
 # "wandb login" in a terminal before using this script
 # Runs states : finished, running, crashed

src/train-corset.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import sys

src/train.py

@@ -1,3 +1,16 @@
+# © - 2024 Université de Mons, Multitel, Université Libre de Bruxelles, Université Catholique de Louvain
+
+# CIA is free software. You can redistribute it and/or modify it 
+# under the terms of the GNU Affero General Public License 
+# as published by the Free Software Foundation, either version 3 
+# of the License, or any later version. This program is distributed 
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
+# without even the implied warranty of MERCHANTABILITY or FITNESS 
+# FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License 
+# for more details. You should have received a copy of the Lesser GNU 
+# General Public License along with this program.  
+# If not, see <http://www.gnu.org/licenses/>.
+
 import hydra
 import os
 import sys