From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id 453334983F for ; Tue, 20 Feb 2024 04:21:03 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id E2C6868D445; Tue, 20 Feb 2024 06:21:00 +0200 (EET) Received: from mgamail.intel.com (mgamail.intel.com [198.175.65.15]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id AA7A068D434 for ; Tue, 20 Feb 2024 06:20:53 +0200 (EET) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=intel.com; i=@intel.com; q=dns/txt; s=Intel; t=1708402859; x=1739938859; h=from:to:subject:date:message-id:references:in-reply-to: content-transfer-encoding:mime-version; bh=HXxWtLI7YEZ1pPE0Lt2DQ2EtoBUdQkuI5H2V9Rb/YLo=; b=AiZDZ3vEHi9D46xX6Tm5eIpw/1ATkKIAZj/NxZPvO1IADIlNZhUDjuOT 5yi6JBkKTN6Z49dPAbb+GnmEM+FjT/L18kcbDtMJ9hSG2J3L2/tHu8xOH sHkvS2ihiCKgvtS34Y4AScQhinYbI9+Ny830otnzL7gse7tfYXKXzgUS1 /IL3UzLaIgkImVDfGhX2hTjEyvZdSUD81AdXz9hswDG9hTl+3Eyaci9kS 3y8RjcR0HwCDcBwVk1JfpVKhmVvQfIHnGEpoFLVUp1xYJs8ygdtowfp/l VjnsCUJfmj1zV74wrTZcd/ijzlV7BOqPHSsZ3PRs7ZuSyhTOnyCv/vpA/ w==; X-IronPort-AV: E=McAfee;i="6600,9927,10989"; a="6315391" X-IronPort-AV: E=Sophos;i="6.06,171,1705392000"; d="scan'208";a="6315391" Received: from fmviesa008.fm.intel.com ([10.60.135.148]) by orvoesa107.jf.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 19 Feb 2024 20:20:52 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="6.06,171,1705392000"; d="scan'208";a="4813231" Received: from fmsmsx601.amr.corp.intel.com ([10.18.126.81]) by fmviesa008.fm.intel.com with ESMTP/TLS/AES256-GCM-SHA384; 19 Feb 2024 20:20:51 -0800 Received: from fmsmsx610.amr.corp.intel.com (10.18.126.90) by fmsmsx601.amr.corp.intel.com (10.18.126.81) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2507.35; Mon, 19 Feb 2024 20:20:51 -0800 Received: from FMSEDG603.ED.cps.intel.com (10.1.192.133) by fmsmsx610.amr.corp.intel.com (10.18.126.90) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2507.35 via Frontend Transport; Mon, 19 Feb 2024 20:20:51 -0800 Received: from NAM11-CO1-obe.outbound.protection.outlook.com (104.47.56.169) by edgegateway.intel.com (192.55.55.68) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.1.2507.35; Mon, 19 Feb 2024 20:20:50 -0800 ARC-Seal: i=1; a=rsa-sha256; s=arcselector9901; d=microsoft.com; cv=none; b=ImwX1oGP6XzX/GMYNkb6XPAv3wSCnIwm3tYOLJLz8rRVa+O4f5Kg155Yho67xXv9z2eVPgr32ToBCDbUb/Oqa1wem75A9dDhcXIRqCv/+I74hhhIt4MckpoqxhXL4EK9DsoK2idAzmTAKU+RyFZyqO9MWQa+eitFGNS7pXN9qqUONIQzct0HhsZ2TeVwJnqD3/gqd16+FD3IT47P/u28ufETU0cRsv27Z6Xq2ufPO9VnL6v+tv6C3ETaKNV8cu54rBOx8yjh+MdmAF48DI1g5XBFMnIepM4q1yG/36ppxA56rgaxO/j//9Nrc8yKA5OjVfjfFBQPvDjIPSDXOWTFcw== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=microsoft.com; s=arcselector9901; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-AntiSpam-MessageData-ChunkCount:X-MS-Exchange-AntiSpam-MessageData-0:X-MS-Exchange-AntiSpam-MessageData-1; bh=wczbuEqYDYztmpCIUPr5mlb0CyBMpNy9A+RUjsszV0Q=; b=jDdSKCHGv5kgXalFadrSrw4nqRAN0UmbnCpjV87cwAV63sUaL+2EvUSjMVsbqq+wZgWxTXnrQM966v9Hputy3OYD9ps08IvzUchEANSxieSUoWnfQSIBKteGo2wW2ViSmwZYisO3+hzlsyGe8cFkcvllVkyyM/C9YsDizYHXC2GYz3uB6lCleCmOYG1BaJYgWQLTvTkplipGPKA5BNWgbqsFrKwJysHS5ES9/7gbRKRGOtVLmDJgrmiZedT/VY6RXxXH5rXLoJ6Qh2c2DfKm5+y6I41WW+8slM7sT/brAkPqkNF4Xm/cw3hIZqvon/EyEjp8fVd7HDMyEMwrnQchlA== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=pass smtp.mailfrom=intel.com; dmarc=pass action=none header.from=intel.com; dkim=pass header.d=intel.com; arc=none Received: from IA1PR11MB6396.namprd11.prod.outlook.com (2603:10b6:208:3ab::10) by SA3PR11MB7611.namprd11.prod.outlook.com (2603:10b6:806:304::17) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.7292.26; Tue, 20 Feb 2024 04:20:48 +0000 Received: from IA1PR11MB6396.namprd11.prod.outlook.com ([fe80::204e:5433:6d25:bde0]) by IA1PR11MB6396.namprd11.prod.outlook.com ([fe80::204e:5433:6d25:bde0%6]) with mapi id 15.20.7292.036; Tue, 20 Feb 2024 04:20:48 +0000 From: "Chen, Wenbin" To: FFmpeg development discussions and patches Thread-Topic: [FFmpeg-devel] [PATCH v2] libavfi/dnn: add LibTorch as one of DNN backend Thread-Index: AQHaVak5eneTUGTDGkKt2F8vPj0ybLEI8j4AgAnET0CAAAYBkA== Date: Tue, 20 Feb 2024 04:20:48 +0000 Message-ID: References: <20240202072629.1259512-1-wenbin.chen@intel.com> In-Reply-To: Accept-Language: en-US Content-Language: en-US X-MS-Has-Attach: X-MS-TNEF-Correlator: authentication-results: dkim=none (message not signed) header.d=none;dmarc=none action=none header.from=intel.com; x-ms-publictraffictype: Email x-ms-traffictypediagnostic: IA1PR11MB6396:EE_|SA3PR11MB7611:EE_ x-ms-office365-filtering-correlation-id: b33593c7-84c7-4e9f-b606-08dc31cb50de x-ms-exchange-senderadcheck: 1 x-ms-exchange-antispam-relay: 0 x-microsoft-antispam: BCL:0; x-microsoft-antispam-message-info: b56n34+oxL/f2BkD4jPYTw+b/8YwIFR72j9oMVtGCP4k2Ygt2b7Esaw86+q6KgsEfLKMAAOWL3AeqEnS6G4rLilQkGepkZGVzhTGfzdly+bjLps7ZLlkzwmQZgesS3P1nx6aA7cdkm/QQUh4Lwc7IeK2PnWdkYMlwLCm5GaQjJOeSSf6zsEarY2diSJAFfuaVbHjDhlalPrqVA1LowVT2+3WpiUxKluFi7iLzU0sk3Ejlax0dNYq9gNBPegVyKzDdelE74vfiuQEcDlT+BdTsmqhjGQzSo9odm6y1z/wqIXBhBqYEeLNCtD5mQztXVabwtDYZa3jpgJHt00FQ3kGBOEklOmSrxdzE+DMRm7kMib+y+AGSRAZR24cfQvECxhPV718dmnZgjY9mRWaA6gOzLS4Xm86yM70Htzax2A/g9lpbW/e8v4m0psKWoYVrj7/MbYZ7XBzGDOVICojEI+5Ref11kwnp6TSRhmPbGYKpC4XyUgooJDCbUYe+COJyViiZvnlCBL4jKPIZVChQfqVn5JOj6UrCVHAdZjMQVIpnbkGtNBhw1n6/yEJNKtfW/9DKBD90UvWCI2smJJiZpTjybVCsNhjnjQXZYfOYa6k4Hs= x-forefront-antispam-report: CIP:255.255.255.255; CTRY:; LANG:en; SCL:1; SRV:; IPV:NLI; SFV:NSPM; H:IA1PR11MB6396.namprd11.prod.outlook.com; PTR:; CAT:NONE; SFS:(13230031)(38070700009); DIR:OUT; SFP:1101; x-ms-exchange-antispam-messagedata-chunkcount: 1 x-ms-exchange-antispam-messagedata-0: =?us-ascii?Q?pf26mFfLoEEpm6dp81VOP5ziQfXl6smHOwEjdG1ybiTo5GPF9Kkt2V6u1OUz?= =?us-ascii?Q?HYdyeK8qhn8rEla8DYWQnxgtnP0m7bb/BRgC13wCXu1QY06ok9c1Yi6GnM3S?= =?us-ascii?Q?9gvQ+EJufwYNRKXnLt87GX40y36cpRwc+raM5FP5H9JUfCdSyBd3l+T0MfmQ?= =?us-ascii?Q?sUlJM2TT25NrqDw5qtcEOExiQuNivsuq+Yb2ShcnV4cyzr+hTyq4HURaE5n/?= =?us-ascii?Q?RBx/IoI4zfEwj9bRJCrrQSGYbmPeP6BRe/0SmptePhjUdKJ3MlSO/FckThiB?= =?us-ascii?Q?MnXN2cjU8dxe41eS8skUes8DburOyGkGBqcWMN/7bHN9sUwG7wPIPnFf+0GU?= =?us-ascii?Q?JKrhHYv816NM+16GZ/OcyPkAZDhcND0OdmyZA6gUYHZy1UhSrtGhPX0pzOKZ?= =?us-ascii?Q?Aylub0H+mLfUB5Wk6TPi/X5i8NnKtpaRg+b/9wxfIPc2hLJx+SNpdMZD4SJ2?= =?us-ascii?Q?/wcHHDVjSSdwuc4/3kn78LfGloOZultP5gjv44lQV8d99SvYmfFEclShOI2X?= =?us-ascii?Q?RR5oRQfetVSAkmhUkjNNi0xMElPkNevC/sYQT4mk+ym6oASttjf34fir6aEC?= =?us-ascii?Q?PzlTPxqgqd2qnHlcJTfq4KVGS+/nH9vS+ROqLX499Oay4VHwepdKdb1XVgur?= =?us-ascii?Q?vrhz+XLlILZxl7ThU8YS4EcqnYayDYugiJIWEMl7AipuPn5XEk/CqOYPQGN7?= =?us-ascii?Q?jecxRCexjnVuWPOsNkrZgcHJwbB8TjZyHIqbGudqJr+/DfdFwVHxPDrstofv?= =?us-ascii?Q?e9IiIJ4qzJqsUpMRwYQhBgIPBz8ohKS6pyyZm3ONAYGb8pjLRYymIKgAIrpA?= =?us-ascii?Q?qtmaBs2DYNMtOVBjUS00bcKpeDxAY2udpN+ZVsKcoeqgwgE+LxgNEf486QjQ?= =?us-ascii?Q?8wJEccJ8Rhlid0Yv+J1stMvCJZwGEAGb66Qy4s+yTSP1MFfHe2zTcTmCDw3A?= =?us-ascii?Q?MvmyPH4VENyq+Sa0vWBgXbmAi3bRt4w2AotZC2Jw/HDQik4zWtj2lY7PNnlg?= =?us-ascii?Q?05hOyAHIAWGZZCu/jpoROPax4PKnz5RSaNb8ARhZGYQL8K8Ml9Ramwv449dq?= =?us-ascii?Q?E8yr99KOWajoQLAN9cRT426WRKbzDvMogBQE26wxx266xDrZc/oH3MUNfU/L?= =?us-ascii?Q?ZZpNW+ROeazSUTelIqihFo430hWxU4JbDk0LVwbr31oGe+z4tjGbZMy+S+K3?= =?us-ascii?Q?YJ7zkf3tKYeMqnKchk6rFNRyA1B8Qdm0Y6IlEmpcUSF7VZ3x6IPSrwVTH05c?= =?us-ascii?Q?9vc6OTNSofotwNV3AV6fCWmR7LwbNV35UYPlQ4C82v0vqUCHgOxGK1T2ucqw?= =?us-ascii?Q?rXakJ2jSfljpbVxM83kTmwT1a8fWnHBZAXkrVmkMonkn/KZ3fsjHKhd+LCk1?= =?us-ascii?Q?qc2eLd3K6yds5r173eE3pWXP8qXLNuCUQBHjmFQW7RUzpTpDBme8ryb+YguB?= =?us-ascii?Q?DYUADhi3KdGMLpgPc0xfS0iySsuP1fBj9c/UI8yO58LzkuXs6L7/euBlY2np?= =?us-ascii?Q?qRzXonALK1Ouv0rrz05iJ4SRys7hgAd8D6KWqmyCcHmabOqUFZnSbKgfwEGI?= =?us-ascii?Q?8jTHGuBAgUiXeoEyhMcUACGGS8/Z+R01lfpqePUG?= MIME-Version: 1.0 X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-AuthSource: IA1PR11MB6396.namprd11.prod.outlook.com X-MS-Exchange-CrossTenant-Network-Message-Id: b33593c7-84c7-4e9f-b606-08dc31cb50de X-MS-Exchange-CrossTenant-originalarrivaltime: 20 Feb 2024 04:20:48.4044 (UTC) X-MS-Exchange-CrossTenant-fromentityheader: Hosted X-MS-Exchange-CrossTenant-id: 46c98d88-e344-4ed4-8496-4ed7712e255d X-MS-Exchange-CrossTenant-mailboxtype: HOSTED X-MS-Exchange-CrossTenant-userprincipalname: 5rAkTVNZXqVXKT5C5ijY4DcteLwYqBqGWRNPIm95G9/gxWqnq1Nxjx3hRUAFBdJ58E/Oc2hfDJOf2pipCE2LWg== X-MS-Exchange-Transport-CrossTenantHeadersStamped: SA3PR11MB7611 X-OriginatorOrg: intel.com Subject: Re: [FFmpeg-devel] [PATCH v2] libavfi/dnn: add LibTorch as one of DNN backend X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: 7bit Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: > > Hello, > > > > On Fri, 2 Feb 2024, at 08:26, wenbin.chen-at-intel.com@ffmpeg.org wrote: > > > +static void infer_completion_callback(void *args) { > > > + THRequestItem *request = (THRequestItem*)args; > > > + LastLevelTaskItem *lltask = request->lltask; > > > + TaskItem *task = lltask->task; > > > + DNNData outputs = { 0 }; > > > + THInferRequest *infer_request = request->infer_request; > > > + THModel *th_model = (THModel *)task->model; > > > + torch::Tensor *output = infer_request->output; > > > + > > > + c10::IntArrayRef sizes = output->sizes(); > > > + assert(sizes.size == 5); > > > > Why 5? > > 5 means 5 channels: [batch_size, frame_number, channel, height, width] Sorry, I mean 5 dimensions. > I only add video SR support, so it only support this type of data for now. > I will change the code to be more easy to read. > > > > > > + outputs.order = DCO_RGB; > > > + outputs.layout = DL_NCHW; > > > + outputs.dims[2] = sizes.at(3); > > > + outputs.dims[3] = sizes.at(4); > > > + outputs.dt = DNN_FLOAT; > > > + outputs.dims[1] = 3; > > > > Why 3? > > It is RGB so the channel is 3, but I should use sizes.at(2) instead of a magic > number. > Thanks for pointing it out. I will update it in patch v3. > > > > > > > -- > > Jean-Baptiste Kempf - President > > +33 672 704 734 > > _______________________________________________ > > ffmpeg-devel mailing list > > ffmpeg-devel@ffmpeg.org > > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > > > To unsubscribe, visit link above, or email > > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe". > _______________________________________________ > ffmpeg-devel mailing list > ffmpeg-devel@ffmpeg.org > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > To unsubscribe, visit link above, or email > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe". _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".