This repository contains scripts and initial assets for Oomplet Stimulus Generation + Match/NoMatch classifications.
Originally Developed by Team HAI Fall 2022 Angela Zhang, Constanza Tong, Ruizi Wang, Yuan Tan
The provided scripts and all related assets fall under the CC BY-NC-SA 4.0 License All future derivations of this code should contain the above attribution
- Set up a Python virtual environment in the root directory
% python -m venv venv
- Activate the virtual environment based on OS
- Install all the requirements
% pip install -r requirements.txt
usage: python generate.py [-h] [-n N] [-p] [-c C] [-v] [-k] [-s S]
options:
-h, --help show this help message and exit
-n N number of oomplets to generate N, by default will generate all possible combinations
-p multiprocessing flag, default off
-c C max number of processes to spawn if multiprocessing, default 4
-v verbose, default off
-k keep existing files in output folder, default off
-s S seed value for randomly generated oomplets, default 0
usage: python categorize.py [-h] [-k]
options:
-h, --help show this help message and exit
-d, --def define your 'Match' criteria
-k keep existing files in output folders, default off
-i, --input name of the directory from which Oomplets will be sorted (default: Oomplets)
-a, --any flags Oomplets with ANY of the given defining attributes as 'Match' (default: off, must have ALL attributes)
Valid attribute options are as follows. Must be typed exactly as shown.
'color_cool', 'color_warm',
'shape_sharp', 'shape_mixed', 'shape_round',
'lash_yes', 'lash_no',
'wide_eyes', 'middle_eyes', 'narrow_eyes',
'short_legs', 'middle_legs', 'long_legs',
'feet_left', 'feet_right', 'feet_in', 'feet_out',
'open_mouth', 'closed_mouth',
'dots_pattern', 'stripes_pattern',
'right_arm_down', 'right_arm_up', 'left_arm_down', 'left_arm_up'