2021年9月4日
opencv 调用 pytorch训练的resnet模型
摘要： 使用OpenCV的DNN模块调用pytorch训练的分类模型，这里记录一下中间的流程，主要分为模型训练，模型转换和OpenCV调用三步。 一、训练二分类模型 准备二分类数据，直接使用torchvision.models中的resnet18网络，主要编写的地方是自定义数据类中的__getitem__， 阅读全文
posted @ 2021-09-04 21:33 半夜打老虎 阅读(3240) 评论(0) 推荐(2)