robotika / osgar

Open Source Garden (Autonomous) Robot
MIT License
13 stars 12 forks source link

Recognize artifacts using a neural network #331

Open zwn opened 4 years ago

zwn commented 4 years ago

There is an older attempt to bring DNN artifact detection started at https://github.com/robotika/subt-artf/tree/master/model. This issue is meant to track information related to that effort.

zwn commented 4 years ago

https://en.wikipedia.org/wiki/SqueezeNet

SqueezeNet was originally described in a paper entitled "SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size." AlexNet is a deep neural network that has 240MB of parameters, and SqueezeNet has just 5MB of parameters.

Model compression (e.g. quantization and pruning of model parameters) can be applied to a deep neural network after it has been trained. In the SqueezeNet paper, the authors demonstrated that a model compression technique called Deep Compression can be applied to SqueezeNet to further reduce the size of the parameter file from 5MB to 500KB.

zwn commented 4 years ago
zwn commented 4 years ago

pytorch and tensorflow are on its way to our base image