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.

Sign up for GitHub

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

Jump to bottom

Image classification using DNNs and Transfer Learning. #4057

Merged
merged 38 commits into from Aug 6, 2019
Merged

Image classification using DNNs and Transfer Learning. #4057

merged 38 commits into from Aug 6, 2019

Conversation

codemzs
Copy link
Member

@codemzs codemzs commented Aug 2, 2019 •
edited

  • Rewired Microsoft.ML.Tensorflow to use Tensorflow .NET C# bindings.
  • Introduces Microsoft.ML.Dnn package that contains transfer learning based image classification API using resnet and inception V3 models.
    • Contains API to retrain a DNN model using tensorflow.

All tensorflow tests are passing. The current test failures have to do with ONNX tests that seem to be using a higher version of protobuf since TF.NET also uses protobuf.

CESARDELATORRE
CESARDELATORRE approved these changes Aug 3, 2019
View reviewed changes
Copy link
Contributor

@CESARDELATORRE CESARDELATORRE left a comment •
edited

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm really excited to try this with more elaborated samples as soon as we have the NuGet package. I provided some feedback with suggestions and possible improvements, but I consider this code could already be merged and keep evolving it from there taking my comments into account, since this still will be a very early preview.

The sooner we can create samples and code like what users would be creating, the better for being able to provide feedback and evolve accordingly. 👍

codemzs and others added 15 commits August 2, 2019 19:08
@Oceania2018
Copy link

Oceania2018 commented Aug 5, 2019 •
edited by codemzs

@CESARDELATORRE We will add more other Model Architectures like MobileNet, ResNet and NASNet, which could give you some extra precision. I will implement it in TF.NET first. Hope it be true for ML.NET v1.4 release. #Resolved

CESARDELATORRE
CESARDELATORRE reviewed Aug 5, 2019
View reviewed changes
src/Microsoft.ML.Dnn/DnnCatalog.cs

if (!File.Exists(options.ModelLocation))
{
if (options.Arch == Architecture.InceptionV3)
Copy link
Contributor

@CESARDELATORRE CESARDELATORRE Aug 5, 2019 •
edited by codemzs

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is great, I see you added InceptionV3 over the weekend! 👍 #Resolved

@codemzs
Copy link
Member Author

codemzs commented Aug 6, 2019

@Oceania2018 This release we are not advertising GPU support. I want to first test it before we remove this dependency.

@codemzs
Copy link
Member Author

codemzs commented Aug 6, 2019

Thanks, @CESARDELATORRE for reviewing.

@codemzs codemzs merged commit 8d51eee into dotnet:master Aug 6, 2019
@CESARDELATORRE CESARDELATORRE mentioned this pull request Aug 7, 2019
Closed
@dotnet dotnet locked as resolved and limited conversation to collaborators Mar 20, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers

@CESARDELATORRE CESARDELATORRE CESARDELATORRE approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
3 participants
@codemzs @Oceania2018 @CESARDELATORRE