02 2021 档案

摘要:torchvision.models 里包含了许多模型,用于解决不同的视觉任务:图像分类、语义分割、物体检测、实例分割、人体关键点检测和视频分类。 本文将介绍 torchvision 中模型的入门使用,一起来创建 Faster R-CNN 预训练模型,预测图像中有什么物体吧。 import torc 阅读全文
posted @ 2021-02-26 09:37 GoCodingInMyWay 阅读(417) 评论(0) 推荐(0) 编辑
摘要:准备数据 准备 COCO128 数据集,其是 COCO train2017 前 128 个数据。按 YOLOv5 组织的目录: $ tree ~/datasets/coco128 -L 2 /home/john/datasets/coco128 ├── images │ └── train2017 阅读全文
posted @ 2021-02-24 10:30 GoCodingInMyWay 阅读(388) 评论(0) 推荐(0) 编辑