HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Add BGR to RGB conversion in LibTorch example (#17864)

Browse source
This commit is contained in:
Mohammed Yasin 2024-11-29 05:50:04 +08:00 • committed by GitHub
parent 4a5dbc0bfb
commit e668de6e02
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
examples/YOLOv8-LibTorch-CPP-Inference/main.cc
View file

@ -226,6 +226,7 @@ int main() {
cv::Mat image = cv::imread("/path/to/bus.jpg"); cv::Mat image = cv::imread("/path/to/bus.jpg");
cv::Mat input_image; cv::Mat input_image;
letterbox(image, input_image, {640, 640}); letterbox(image, input_image, {640, 640});
cv::cvtColor(input_image, input_image, cv::COLOR_BGR2RGB);
torch::Tensor image_tensor = torch::from_blob(input_image.data, {input_image.rows, input_image.cols, 3}, torch::kByte).to(device); torch::Tensor image_tensor = torch::from_blob(input_image.data, {input_image.rows, input_image.cols, 3}, torch::kByte).to(device);
image_tensor = image_tensor.toType(torch::kFloat32).div(255); image_tensor = image_tensor.toType(torch::kFloat32).div(255);