search

chail / patch-forensics

Investigating patches for fake image classification
130 stars 18 forks source link

some questions on release code? #1

Closed zj19921221 closed 3 years ago

zj19921221 commented 4 years ago

hi I met a question when i read the script of "train.py"

In "train.py " of Line 3 "from models import create_model" there is no file named "create_model.py"?

Was you missed it ?

looking forward your reply!

chail commented 4 years ago

Hi,

The function create_model is defined in models/__init__.py (here). It will choose whether to use the basic discriminator or patch discriminator based on the options you provide, and then within basic/patch discriminator it will initialize the discriminator architecture from the provided options.

I hope this helps!