/CNN_Toolbox

Here are some interesting scripts used in CNN.

Primary LanguagePython

CNN_Toolbox

Introduction

Here are some useful scripts for CNN implementation. These scripts are mainly based on Caffe library.

Summary

Please specify the path of caffe in caffe_path.txt first!

  • Model Complexity

    • This script is for FLOPs (FLoating-point OPerations) & parameters calculation.

    • You need add the *.prototxt file in this folder, and run:

        python FLOPs_and_size.py

Acknowledge

These scripts are collected from Internet. However, we cannot send our gratitude one by one. Here, we thank everyone who made contribution to this toolbox.