计算机视觉教程：用OpenCV与PyTorch快速完成图像预处理和ResNet18模型微调

import cv2 img = cv2.imread("demo.jpg") print(img.shape) gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) cv2.imshow("gray", gray) cv2.waitKey(0) cv2.destroyAllWindows()

import torch import torchvision.models as models import torch.nn as nn model = models.resnet18(weights="IMAGENET1K_V1") for p in model.parameters():

p.requires_grad = False

model.fc = nn.Linear(model.fc.in_features, 10) optimizer = torch.optim.Adam(model.fc.parameters(), lr=1e-3)