30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
import os
|
|
import shutil
|
|
|
|
def mkdir(dirname):
|
|
if not os.path.isdir(dirname):
|
|
os.mkdir(dirname)
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
|
types = ['fire', 'water', 'electric', 'rock', 'grass', 'ice', 'steel']
|
|
animals = ['cat', 'dog', 'rat', 'rabbit', 'dragon', 'duck', 'turtle', 'butterfly', 'monkey', 'bee', 'fox', 'flower', 'horse', 'jellyfish', 'snake', 'Tyrannosaurus', 'dinosaur', 'fish', 'whale', 'bat', 'bear', 'deer', 'pig', 'eagle', 'chicken']
|
|
|
|
|
|
mkdir('./output')
|
|
for type_str in types:
|
|
for animal in animals:
|
|
mkdir('./output/{}_{}'.format(type_str, animal))
|
|
for index in range(1):
|
|
mkdir('./output/{}_{}/{}'.format(type_str, animal, index))
|
|
prompt = "a {} type pokemon, it looks like a {}".format(type_str, animal)
|
|
print("PROMPT: ", prompt)
|
|
|
|
os.system('python dreamgaussian/main.py --config dreamgaussian/configs/text_mv.yaml prompt="{}" save_path=output'.format(prompt))
|
|
os.system('python dreamgaussian/main2.py --config dreamgaussian/configs/text_mv.yaml prompt="{}" save_path=output'.format(prompt))
|
|
|
|
for file in os.listdir('./logs'):
|
|
shutil.move('./logs/{}'.format(file), './output/{}_{}/{}/{}'.format(type_str, animal, index, file))
|
|
|