-layer_reorg_src = files('reorg_layer.cpp')
+# build command for lib_yolov2_loss_layer.so
+yolov2_loss_src = files('yolo_v2_loss.cpp')
+yolov2_loss_layer = shared_library('yolov2_loss_layer',
+ yolov2_loss_src,
+ dependencies: [nntrainer_dep, nntrainer_ccapi_dep],
+ include_directories: include_directories('./'),
+ install: true,
+ install_dir: nntrainer_libdir/'nntrainer'/'layers',
+ cpp_args: '-DPLUGGABLE'
+)
+yolov2_loss_layer_dep = declare_dependency(
+ link_with: yolov2_loss_layer,
+ include_directories: include_directories('./')
+)
# build command for lib_reorg_layer.so
+layer_reorg_src = files('reorg_layer.cpp')
reorg_layer = shared_library('reorg_layer',
layer_reorg_src,
dependencies: [nntrainer_dep, nntrainer_ccapi_dep],
include_directories: include_directories('./'),
install: true,
- install_dir: application_install_dir,
+ install_dir: nntrainer_libdir/'nntrainer'/'layers',
cpp_args: '-DPLUGGABLE'
)
-
reorg_layer_dep = declare_dependency(
link_with: reorg_layer,
include_directories: include_directories('./')
yolo_dependencies = [app_utils_dep,
nntrainer_dep,
nntrainer_ccapi_dep,
+ yolov2_loss_layer_dep,
reorg_layer_dep
]
*/
#include "yolo_v2_loss.h"
-#include <iostream>
+#include <nntrainer_log.h>
namespace custom {
float class_loss = mse(masked_class_pred, masked_class_gt);
float loss = 5 * bbox_loss + confidence_loss + class_loss;
- std::cout << "\nCurrent iteration loss: " << loss << std::endl;
+ ml_logd("Current iteration loss: %f", loss);
}
void YoloV2LossLayer::calcDerivative(nntrainer::RunLayerContext &context) {