how to generate the img_sal_sub image path list, I cannot found any code about generate?

CyberAgentAILab / RALF

[CVPR24 Oral] Official repository for RALF: Retrieval-Augmented Layout Transformer for Content-Aware Layout Generation

https://arxiv.org/abs/2311.13602

Apache License 2.0

96 stars 1 forks source link

how to generate the img_sal_sub image path list, I cannot found any code about generate? #16

Closed trouble-maker007 closed 3 months ago

trouble-maker007 commented 3 months ago

@UdonDa img_sal = Image.open(self.sal[idx]).convert("L") img_sal_sub = Image.open(self.sal_sub[idx]).convert("L") img_sal_map = Image.fromarray(np.maximum(np.array(img_sal), np.array(img_sal_sub)))

UdonDa commented 3 months ago

I do not remember the structure of the folder actually. But, I think saliency means ISNet and saiency_sub means BASNet. You can modify the name of the directory. https://github.com/CyberAgentAILab/RALF/blob/main/image2layout/hfds_builder/saliency_detection.py#L105