YOLOv5代码详解(test.py部分)
目录 2. test.py 2.1 设置超参数 2.2 设置任务(验证,测试,学习) 2.3 测试函数 2.3.1 初始化模型 2.3.2 判断设备类型并仅使用一张GPU进行测试 2.3.3 获取配置文件路径和文件参数 2.3.4 数据获取 2.3.5 计算map数据 2.3.6 打印结果(图片,速度),保存结果至json,并返回结果 2. test.py 该部分主要用于运行train.py时,计算每个epoch的mAP。 PS,与train.py相似的部分就不再阐述。 2.1 设置超参数 权重,数据,batch size,图像尺寸,使用哪张显卡,数据增强,计算mAP。if __name__ == '__main__': parser = argparse.ArgumentParser(prog='test.py') parser.add_argument('--weights', type=str, default='weights/best.pt', help='model.pt path') parser.add_argument('--data', type=str, default='data/coco.yaml', help='*.data path') parser.add_argument('--batch-size', type=int, default=16,