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 }