Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Request more support on Tensorflow format model #153

Open
NeilAlfred opened this issue May 30, 2024 · 0 comments
Open

Request more support on Tensorflow format model #153

NeilAlfred opened this issue May 30, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@NeilAlfred
Copy link

First of all, I am a fan of the Tensorflow framework, so I was very happy when I found that ModelScan supports security detection of Tensorflow models(Protocol Buffer format). This is great for me.

However, I carefully read the spec and source code of ModelScan, and I found that ModelScan's support for the Tensorflow framework seems to be a bit limited. It seems to only support the API detection of ReadFile, WriteFile, Lambda Layer and tf.io.matching_files.

ModelScan's support for Pytorch is very beautiful and elegant, so I am curious if ModelScan has further plans to improve its support for Tensorflow? Or are there other API calls in Tensorflow that will cause Model Serialization Attack?

@NeilAlfred NeilAlfred added the enhancement New feature or request label May 30, 2024
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant