Pytorch预训练集进行训练测试

首先主题:使用pytorch的预训练集来进行训练测试;安装配置环境:直接在GPU上跟着先跑通一遍,能当成一个调包侠。主要是安装一些库,其中最基础的库:mmcv来做底层的运算。下载1000多个类别分类数据集。
B1是核心:导入基础包;找计算设备;导入预训练图像分类模型—选一个reset18模型(可换层数更多的模型)。步骤:1.图像预处理:RSTN。R—resize;载入图像,pilow包;执行图像分类预测,看每个类别的概率-置信度,图像分类结果写在图像上。
视频处理预测:逐帧处理,两种可视化方案。
摄像头训练。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值