JasonDCox / ML-Mentorship-GovSchool

0 stars 0 forks source link

Create Specific Tensorflow pet identification on Desktop #29

Closed brandonC1234 closed 2 years ago

brandonC1234 commented 2 years ago

Description

Create a Tensorflow model in which I can upload images of my dogs with annotations to a folder and it will train to identify the different dogs. This model with work with the first one to account for the large amounts of false positives it will create from limited data input.

Acceptance Criteria

References

brandonC1234 commented 2 years ago

Progress

I have a system where you take images through your webcam and use the provided annotation software to annotate them. You then have to change the label map accordingly and create a TFrecord with it. This allows you to train the model.

Performance

It does detect the dog, but it has an issue where it gives a lot of false positives. To address this, I will be crosschecking the results with a model trained to detection general cats and dogs.

brandonC1234 commented 2 years ago

Changes

I changed to the strategy of having a class for each breed of dog and a class for the unique dog.

Progress

After 74,000 training steps, the model is complete and works properly

Performance

The performance and metrics for the model has been uploaded in the Tensorflow folder in the metrics_results folder

FPS and memory usage were not measured as they are not important to the project since this is being run on the desktop rather than the Jetson.