chapter7 import cv2 import numpy as np from cv2tools import stackImages def fun(x): pass cv2.namedWindow("bars") cv2.resizeWindow("bars", 500, 300) cv2.createTrackbar("hue min", "bars", 0...

chapter2 import cv2 import numpy as np img = cv2.imread('Resources/lena.png') gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # cv2.GaussianBlur 第三个参数表示高斯的标准差 为0代表根据kernel size自动计算 blur_img = ...

chapter4 import cv2 import numpy as np img = np.zeros((512,512,3), np.uint8) # img[:] = 255,0,0 cv2.line(img, (0,0), (img.shape[1],img.shape[0]), (0,255,0), 3) cv2.rectangle(img, (0,0), (200,300...

model structure class CNNBlock(nn.Module): def __init__(self, in_channels, out_channels, kernel_size, padding=0, stride=1): super().__init__() self.conv = nn.Conv2d(in_channels...

loss function code import torch import torch.nn as nn from utils import intersection_over_union def intersection_over_union(boxes_preds, boxes_labels, box_format="midpoint"): """ This f...

dataset code ANCHORS = [ [(0.28, 0.22), (0.38, 0.48), (0.9, 0.78)], [(0.07, 0.15), (0.15, 0.11), (0.14, 0.29)], [(0.02, 0.03), (0.04, 0.07), (0.08, 0.06)], ] # Note these have been r...

Yolo-v3 Model structure Yolo v3 Model Structure Code Snippet Shape of prediction and target prediction head1: torch.Size([bs, 3, S, S, 5+cls]) head2: torch.Size([bs, 3, S...

general yolo model structure class BackBone(nn.Module): def __init__(self, model: nn.Module, extract_ids: list): super().__init__() self.model...

SSD Model Struct

import packages !pip install pycocotools !wget https://raw.githubusercontent.com/pytorch/vision/main/references/detection/engine.py !wget https://raw.githubusercontent.com/pytorch/vision/main/refe...