直接用乔熙的BGR输出

Hu Chunming
1 parent e69cf19f
Showing 2 changed files with 10 additions and 9 deletions
vehicle_structure_platform.git0708-3080-trt-face/src/SfxDecoder/DxDecoderWrap.cpp
vehicle_structure_platform.git0708-3080-trt-face/src/SfxDecoder/SfxDecoder.cpp
@@ -294,6 +294,7 @@ void DxDecoderWrap::DxCUDADecoderCallback( void * buf, unsigned int size, int wi
 			unsigned char *pHwData = nullptr;
 			cudaError_t cudaStatus = cudaMalloc((void **)&pHwData, rgbSize);
 			if (cudaStatus != cudaSuccess) {
+				m_queue_frames_mutex.unlock();
 				LOG_ERROR("[{}]- cudaMalloc failed !!!", m_name.c_str());
 				return ;
 			}
@@ -308,13 +309,13 @@ void DxDecoderWrap::DxCUDADecoderCallback( void * buf, unsigned int size, int wi
 			// unsigned char* cpu_data = new unsigned char[rgb_size];
 			// cudaStatus = cudaMemcpy(cpu_data, pHwData, rgb_size * sizeof(unsigned char), cudaMemcpyDeviceToHost);
  
-			// if (timestamp > 245  && timestamp < 255)
-			// {
-			// 	string file_name = "./nv12/"+to_string(timestamp) + ".txt";
-			// 	FILE* fp=fopen(file_name.c_str(),"wb");
-			// 	fwrite(cpu_data, rgb_size, 1, fp);
-			// 	fclose(fp);
-			// }
+			// // if (timestamp > 245  && timestamp < 255)
+			// // {
+			// // 	string file_name = "./nv12/"+to_string(timestamp) + ".txt";
+			// // 	FILE* fp=fopen(file_name.c_str(),"wb");
+			// // 	fwrite(cpu_data, rgb_size, 1, fp);
+			// // 	fclose(fp);
+			// // }
  
 			// cv::Mat img(m_height, m_width, CV_8UC3, cpu_data);
 			// cv::imwrite("test.jpg", img);
@@ -43,8 +43,8 @@ bool SfxDecoder::OpenDecoder(DxDecoderConfig&amp; cfg)
 	sfx_cfg->flags.useFastDecode = 0;
 	sfx_cfg->flags.accelId = m_cfg.devId;
 	sfx_cfg->flags.frameNumber = 1;
-	sfx_cfg->flags.fmt = SFX_VIDEO_FORMAT_NV12U8;
-	//sfx_cfg->flags.fmt = SFX_VIDEO_FORMAT_C3BGRU8;
+	// sfx_cfg->flags.fmt = SFX_VIDEO_FORMAT_NV12U8;
+	sfx_cfg->flags.fmt = SFX_VIDEO_FORMAT_C3BGRU8;
  
 	m_handler = SfxStreamHandlerInterface::SfxGetInstance(sfx_cfg);