| 49 | return img |
| 50 | |
| 51 | def get_args_parser(): |
| 52 | parser = argparse.ArgumentParser('Double Conditioning LDM Finetuning', add_help=False) |
| 53 | # project parameters |
| 54 | parser.add_argument('--root', type=str, default='../dreamdiffusion/') |
| 55 | parser.add_argument('--dataset', type=str, default='GOD') |
| 56 | parser.add_argument('--model_path', type=str) |
| 57 | |
| 58 | parser.add_argument('--splits_path', type=str, default=None, |
| 59 | help='Path to dataset splits.') |
| 60 | parser.add_argument('--eeg_signals_path', type=str, default=None, |
| 61 | help='Path to EEG signals data.') |
| 62 | |
| 63 | parser.add_argument('--config_patch', type=str, default=None, |
| 64 | help='sd config path.') |
| 65 | |
| 66 | parser.add_argument('--imagenet_path', type=str, default=None, |
| 67 | help='imagenet path.') |
| 68 | |
| 69 | return parser |
| 70 | |
| 71 | |
| 72 | if __name__ == '__main__': |