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