Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

toddwyl/nailtracking

Repository files navigation

Real-time Nail-Detection using Neural Networks (SSD) on Tensorflow.

This repo is similar to the handtracking. I train the nail detector using MobileNetV1-FPN-SSD, and the dataset I used is annotated by laboratory member.

Here is the nail detector in action and draw the line:

nail_draw

The annotated samples like the below:

1560172595651

The overall process is divided into 3 steps:

  • Histogram equalization

  • Skin color binarization(convert the RGB channel to YCbCr and use inRange method to binarize)

  • MobileNet-SSD-FPN

    1560172217358

The MobileNetV1-FPN-SSD architecture may like this below:

ssd_fpn_nail

I use the MobileNet-SSD first, but the nail is too tiny to detect. So we can see the MobileNet-SSD-FPN is pretty better than MobileNet-SSD.

Method mAP mAP(large) mAP(medium) mAP(small) mAP@0.50IOU mAP@0.75IOU
MobileNet-SSD 45.20 62.58 48.14 13.14 75.79 45.81
MobileNet-SSD-FPN 60.01 75.03 64.46 29.25 84.65 71.85

1560173400082

About

Real-time Nail-Detection using Neural Networks (SSD) on Tensorflow.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

Contributors

Languages

AltStyle によって変換されたページ (->オリジナル) /