Function BCELoss

lesson7-Detection/src/models/yolo_training.cpp:150–154

}

torch::Tensor BCELoss(torch::Tensor pred, torch::Tensor target) {
	pred = clip_by_tensor(pred, 1e-7, 1.0 - 1e-7);
	auto output = -target * torch::log(pred) - (1.0 - target) * torch::log(1.0 - pred);
	return output;
}


YOLOLossImpl::YOLOLossImpl(torch::Tensor anchors_, int num_classes_, int img_size_[],