ChunML / ssd-tf2

A super clean implementation of SSD (Single Shot MultiBox Detector) made possible by Tensorflow 2.0
MIT License
115 stars 54 forks source link

How do I create very small SSD for learning how it works? #4

Closed offchan42 closed 4 years ago

offchan42 commented 4 years ago

I don't want any VGG or GiganticResNexT-9000 base model. I want a few conv layers and train it with an easy dataset e.g. tracking a PS joystick that cannot deform or change color like a lizard.

  1. Can you recommend article/course that teaches SSD so that you can implement it from the ground up?
  2. How do I add a SSD with stupid simple base model using this repo?
ChunML commented 4 years ago

Sorry, I was so late. In order to make a small SSD, you must first understand how SSD was created. I recommend that you read the paper a few more times to grasp the idea (no offense, I had to, too).

In short, you don't need that many output layers since their purpose is to learn the features of objects of different sizes. If your data only has small objects (i.e. ~ image_size / 10), you can only use the output from conv4 and conv7 (or just conv4). You may also want to read more about the receptive field to understand the effect of CNN's depth.