amiratag/ACE

support for resnet and custom models

dsvrsec opened this issue · 2 comments

Is there any extra code to be written to test ACE for custom models/resnet

Hi! I am also working on this task with my own models.
I think you can check the original tacv code. https://github.com/tensorflow/tcav/blob/master/tcav/model.py
In the model part, they provide a ModelWrapper class called KerasModelWrapper(ModelWrapper).
So that you can also add a model_wrapper in ace_helpers. In this case you can use your own trained keras model including resnet or others.
Hope this can help you.

ud0y commented

Hi! recently I am trying to work on this task with my own models. But I don't understand the contents inside source_dir folder. Can you guys help me out?