Skip to content

wangqingbaidu/HSNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HSNet

目录结构:

RedNet

|___forward_src 			前馈网络模型代码

	|___pnet.c			动态库入口文件其中封装了模型加载预测等相关接口
	
|___src					RedNet网络核心代码包括模型训练以及预测
	
|___tools				相关脚本

	|___analysis.py			用于多分类然后融合判断正确率的脚本|___assigning_task.py		分配任务脚本|___best_model.py		RedNet本身不支持训练时候保存最好的模型所以这个脚本用来从模型中选择最好的模型|___get_list.py			生成训练数据以及测试数据集脚本|___predict_web_data.py		去年郭老师的预测网络数据代码其中包括预测已经生成相应的报告|___cfg			 		存储网络参数相关配置以及对应的dataset

	|___pnet_list			最近一批数据生成的训练以及测试集|___darknet_S_SC_C.cfg 		其中S对应的是模型的大小es非常小sm小ori原始l大SC对应有无shortcutC代表分类个数
	
	|___*.dataset			RedNet网络参数配置包括训练集测试集list分类中心半停阈值等|___Makefile				生成可以训练以及预测的RedNet可执行程序

|___makelib				生成动态链接库

|___trainNet.sh				训练网络的启动脚本
	

tips:

  1. 所有的脚本通过-h可以输出参数使用说明。

  2. Make文件中EXEC=../train_data/darknet用于执行生成的文件目录。

  3. 生成的可执行程序分类任务使用classifier,训练使用train,验证使用valid,预测使用label。

    例如训练网络使用./darknet -i 0 classifier train pnet.dataset ./cfg/darknet_es_shortcut_22.cfg

Releases

No releases published

Packages

No packages published

Languages