随笔分类 - Python
摘要:图片大小计算 图像分辨率*位深度 位深度一般为32位,R、G、B、α(透明度)各占8位,共4个字节 1080P图像,字节大小为:1920x1080x32(bit) = 66355200bit 66355200bit/8 = 8294400.0B 8294400.0B/1024 = 8100.0KB
阅读全文
摘要:1 torchstat 该工具包可通过pip直接安装: pip install torchstat 使用方法 import torchvision.models as models #pretrained=True就可以使用预训练的模型 #resnet18 = models.resnet18(pre
阅读全文
摘要:模型导出测试 import numpy as np import torch import torch.nn as nn ## 构造随机输入,参数为shape,输入满足正太分布 input = torch.randn(4, 256, 256) input=torch.randint_like(inp
阅读全文
摘要:一 基本使用 1 下载源代码 网址:https://github.com/ultralytics/yolov5 使用git clone到本地。 2 模型训练 使用pycharm打开yolov5文件夹,运行train.py脚本,会自动下载COCO128数据集,如果本地已有数据集,修改coco128
阅读全文
摘要:1 视频解码图片 import cv2 #根据需要修改路径 cap = cv2.VideoCapture("D:/test/test.mp4") i = 0 while cap.isOpened(): ret, frame = cap.read() if not ret: break # frame
阅读全文