support for resnet and custom models
dsvrsec opened this issue · 2 comments
dsvrsec commented
Is there any extra code to be written to test ACE for custom models/resnet
yingji425 commented
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?