search

ChineseYjh / SoftCollage

The official PyTorch implementation of the CVPR2022 paper, SoftCollage: A Differentiable Probabilistic Tree Generator for Image Collage
3 stars 2 forks source link

Model Weights and inference code #1

Open nbardy opened 2 years ago

nbardy commented 2 years ago

Hey. I'm interesting in exploring this model some. I didn't see any mention of inference code in the readme. Looks like training code.

If there is pre-trained models available and inference code I'd appreciate if you could show me.

ChineseYjh commented 2 years ago

Hi. Our method aims to search the optimal tree given the image collection, thus each image collection corresponds to one particular pre-trained model. So we do not provide the pre-trained weights because AIC has too many collections thus all the weights are too large.

For inference codes, you can refer to the following function to write your own inference codes. https://github.com/ChineseYjh/SoftCollage/blob/9c7fd94c0a5e5647d88bc2b3f1de1319dc209795/ops.py#L18