KerwinKai / Caffe_model_zoo

Caffe model zoo, model convert by pre-train pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Caffe_model_zoo

Caffe model zoo, model convert by pre-train pytorch

Start

Activate the conda environment of Python and the corresponding library dependencies

conda create -n py37 python=3.7
conda activate py37
pip install brocolli-caffe
git clone https://github.com/KerwinKai/Caffe2Darknet.git
git clone https://github.com/KerwinKai/Caffe_model_zoo.git

Task

使用Caffe2DarkNet,将已有的Caffe框架下的预训练模型转换至Darknet框架。提供四种轻量级网络以供选择,有以下挑战需要解决克服。

MobileNet_v2

转换工具提示 assert (i + 1 < layer_num and layers[i + 1]['type'] == 'ReLU'),转换后推理未测试

ResNet18

转换成功,推理时需要解决 Flatten在DarkNet上实现的问题

shufflenet_v2

转换工具提示 unknown type Concat,可调研DarkNet上该层的实现方法(能否实现、有无类似功能的层),更新转换工具,转换后推理未测试

squeezenet

转换工具提示 unknown type Concat,可调研DarkNet上该层的实现方法(能否实现、有无类似功能的层),更新转换工具,转换后推理未测试

Related Links

DarkNet Original author website:https://pjreddie.com/darknet/imagenet/

DarkNet wiki:https://github.com/AlexeyAB/darknet/wiki

Python examples using the C API

https://github.com/AlexeyAB/darknet/blob/master/darknet.py

https://github.com/AlexeyAB/darknet/blob/master/darknet_video.py

About

Caffe model zoo, model convert by pre-train pytorch


Languages

Language:Jupyter Notebook 100.0%