Skip to content

MICL-AI/darknet-modified

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

darknet-modified

turn to darkent repository for futher pruning and quantize, but without fp16(half precision) path.

A Darknet branch modified to FP16 excution.
No weights file uploaded.
Run with ./darknet TL predict tiny
in which predict could be valid/predic16/valid16 and nets could be whoever with both cfg and weights files.

Original args still available.

Now the nets are:

  • alexnet
  • mobilenet
  • resnet50
  • squeezenet
  • tiny
  • vgg16

My Modification

  • argvs to simplify the execution

  • fixed the mobilenet -O0 NaN bug with variance[f] = (variance[f] < 0) ? 0 : variance[f];

  • jliu added a entire F16 datapath, I tested and debuged net above

  • move all weights file in ./weights and all cfg file in ./cfg

  • specialized path for mobilenet/squeezenet whos converted form caffe (uses caffe lables)

  • now added a compressed sparse row storage method in connected layers

  • added a feature map pruning, with Makefile PRUNE control to test the dynamic feature map pruning paper here

TODO

  • GPU Dynamic Pruning test

  • Quantization test

Notes

  • see this issue for -O0 compile of YOLOv3

REF

squeezenet-source

convert from here

validating on ImageNet

you should build you image database as

https://pjreddie.com/darknet/imagenet/

then

./darknet classifier valid cfg/imgenet1k.data squeezenet.cfg squeezenet_darknetformat.weights

About

A modified darknet for different precision tests

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages