INNER CODE UNIT · Python
out
junshutang/Make-It-3D · main.py:164
out = blip_model.generate(**inputs)
caption = processor.batch_decode(out, skip_special_tokens=True)[0].strip()
caption = caption.replace("there is ", "")
caption = caption.replace("close up", "photo")
for d in ["black background", "white background"]:
if d in caption:
caption = caption.replace(d, "ground")
print("Caption: ", caption)
opt.text = caption
with open(os.path.join(opt.workspace, 'setting.txt'), 'w') as f:
f.writelines('------------------ start ------------------' + '\n')
for eachArg, value in optDict.items():
f.writelines(eachArg + ' : ' + str(value) + '\n')
f.writelines('------------------- end -------------------')
# only support alpha photo input.