Skip to content
New issue

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

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

Already on GitHub? Sign in 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant