int main(int argc, char **argv)
{
//test_resize("data/bad.jpg");
//test_box();
//test_convolutional_layer();
if(argc < 2){
fprintf(stderr, "usage: %s <function>\n", argv[0]);
return 0;
}
gpu_index = find_int_arg(argc, argv, "-i", 0);
if(find_arg(argc, argv, "-nogpu")) {
gpu_index = -1;
}
#ifndef GPU
gpu_index = -1;
#else
if(gpu_index >= 0){
cudaError_t status = cudaSetDevice(gpu_index);
check_error(status);
}
#endif
if(0==strcmp(argv[1], "imagenet")){
run_imagenet(argc, argv);
} else if (0 == strcmp(argv[1], "average")){
average(argc, argv);
} else if (0 == strcmp(argv[1], "yolo")){
run_yolo(argc, argv);
} else if (0 == strcmp(argv[1], "cifar")){
run_cifar(argc, argv);
} else if (0 == strcmp(argv[1], "go")){
run_go(argc, argv);
} else if (0 == strcmp(argv[1], "rnn")){
run_char_rnn(argc, argv);
} else if (0 == strcmp(argv[1], "vid")){
run_vid_rnn(argc, argv);
} else if (0 == strcmp(argv[1], "coco")){
run_coco(argc, argv);
} else if (0 == strcmp(argv[1], "classifier")){
run_classifier(argc, argv);
} else if (0 == strcmp(argv[1], "tag")){
run_tag(argc, argv);
} else if (0 == strcmp(argv[1], "compare")){
run_compare(argc, argv);
} else if (0 == strcmp(argv[1], "dice")){
run_dice(argc, argv);
} else if (0 == strcmp(argv[1], "writing")){
run_writing(argc, argv);
} else if (0 == strcmp(argv[1], "test")){
test_resize(argv[2]);
} else if (0 == strcmp(argv[1], "captcha")){
run_captcha(argc, argv);
} else if (0 == strcmp(argv[1], "nightmare")){
run_nightmare(argc, argv);
} else if (0 == strcmp(argv[1], "change")){
change_rate(argv[2], atof(argv[3]), (argc > 4) ? atof(argv[4]) : 0);
} else if (0 == strcmp(argv[1], "rgbgr")){
rgbgr_net(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "denormalize")){
denormalize_net(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "normalize")){
normalize_net(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "rescale")){
rescale_net(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "partial")){
partial(argv[2], argv[3], argv[4], atoi(argv[5]));
} else if (0 == strcmp(argv[1], "stacked")){
stacked(argv[2], argv[3], argv[4]);
} else if (0 == strcmp(argv[1], "visualize")){
visualize(argv[2], (argc > 3) ? argv[3] : 0);
} else if (0 == strcmp(argv[1], "imtest")){
test_resize(argv[2]);
} else {
fprintf(stderr, "Not an option: %s\n", argv[1]);
}
return 0;
}
int main()
{
#ifdef OOLUA_LUABIND_COMPARE
# ifdef MFUNC_TEST
{
OOLUA::Script L;
open_Luabind_set_n_get(L);
run_compare(L,"mfunc","LuaBind");
}
# endif
# ifdef VFUNC_TEST
{
OOLUA::Script L;
open_Luabind_hierarchy(L);
run_compare(L,"vfunc","LuaBind");
}
# endif
# ifdef CLASS_PARAM_IMPLICIT_CAST_TEST
{
OOLUA::Script L;
open_Luabind_hierarchy(L);
run_compare(L,"class_param","LuaBind");
}
# endif
#endif
#ifdef OOLUA_LUABRIDGE_COMPARE
# ifdef MFUNC_TEST
{
OOLUA::Script L;
open_LuaBridge_set_n_get(L);
run_compare(L,"mfunc","LuaBridge");
}
# endif
# ifdef VFUNC_TEST
{
OOLUA::Script L;
open_LuaBridge_hierarchy(L);
run_compare(L,"vfunc","LuaBridge");
}
# endif
# ifdef CLASS_PARAM_IMPLICIT_CAST_TEST
{
OOLUA::Script L;
open_LuaBridge_hierarchy(L);
run_compare(L,"class_param","LuaBridge");
}
# endif
#endif
#ifdef MFUNC_TEST
{
OOLUA::Script L;
L.register_class<Set_get>();
run_compare(L,"mfunc","OOLua");
}
#endif
#ifdef VFUNC_TEST
{
OOLUA::Script L;
OOLUA::register_class<ProfileMultiBases>(L);
run_compare(L,"vfunc","OOLua");
}
#endif
#ifdef CLASS_PARAM_IMPLICIT_CAST_TEST
{
OOLUA::Script L;
OOLUA::register_class<ProfileMultiBases>(L);
run_compare(L,"class_param","OOLua");
}
#endif
#if defined OOLUA_SLB_COMPARE && LUA_VERSION_NUM >= 502
# ifdef MFUNC_TEST
{
SLB3::extra::Script L;
L.init();
SLB3::Register<Set_get>(L);
run_compare(L,"mfunc","SLB3");
}
# endif
# ifdef VFUNC_TEST
{
SLB3::extra::Script L;
L.init();
SLB3::Register<ProfileBase>(L);
SLB3::Register<ProfileAnotherBase>(L);
SLB3::Register<ProfileDerived>(L);
SLB3::Register<ProfileMultiBases>(L);
run_compare(L,"vfunc","SLB3");
}
# endif
# ifdef CLASS_PARAM_IMPLICIT_CAST_TEST
{
SLB3::extra::Script L;
L.init();
SLB3::Register<ProfileBase>(L);
SLB3::Register<ProfileAnotherBase>(L);
SLB3::Register<ProfileDerived>(L);
SLB3::Register<ProfileMultiBases>(L);
run_compare(L,"class_param","SLB3");
}
# endif
#endif
#ifdef OOLUA_SWIG_COMPARE
# ifdef MFUNC_TEST
{
OOLUA::Script L;
luaopen_swig_profile(L);
run_compare(L,"mfunc","SWIG");
}
# endif
# ifdef VFUNC_TEST
{
OOLUA::Script L;
luaopen_swig_profile(L);
run_compare(L,"vfunc","SWIG");
}
# endif
# ifdef CLASS_PARAM_IMPLICIT_CAST_TEST
{
OOLUA::Script L;
luaopen_swig_profile(L);
run_compare(L,"class_param","SWIG");
}
# endif
#endif
}
