search

lindsey98 / PhishIntention

PhishIntention: Phishing detection through webpage intention
MIT License
44 stars 12 forks source link

Model Weights and Input Image Sizes #15

Open gyNancy opened 1 year ago

gyNancy commented 1 year ago

Hi,

Are the weights of the models (Siamese, OCR, element recognition) clean-trained or adversarially trained weights? Do input image sizes for the Siamese model have to be 224? Would other input image sizes work for the model?

Thanks!

lindsey98 commented 1 year ago

Hi Nancy, All the models are clean-trained, we didn't observe any phishing pages using adversarial logos. Your original image can be of different sizes but will be resized as 224x224 (https://github.com/lindsey98/PhishIntention/blob/main/phishintention/src/OCR_siamese_utils/inference.py#L50). Since the model is trained with 224x224, it is suggested that the testing images also follow the same format.