mchong6 / SOAT

Official PyTorch repo for StyleGAN of All Trades: Image Manipulation with Only Pretrained StyleGAN.
MIT License
380 stars 56 forks source link

Transfer multiple features from image to image using bbox #12

Open venuv opened 2 years ago

venuv commented 2 years ago

I'd like to annotate and transfer multiple bboxes from the target image to the source image. I noticed in running infinity.ipynb that while I can use Colab to annotate multiple bboxes on the target, only the first of them gets transferred from the source.

This behavior seems consistent with the blend_bbox code in model.py where only the first bbox (coord[0]) is being considered and the rest don't seem to be processed.

Two questions - a) am I missing something here and b) if the above is accurate, is it useful if I add an outer loop to enable blend_bbox to iterate thru the bbox_list