本文只是把关键点部分去掉了,网络部分还没改。
可以用在别的物体检测领域,测试ok
本文是给自己做笔记用的,不喜勿喷。
# -*- coding: UTF-8 -*-
import argparse
import time
from pathlib import Path
import cv2
import torch
import torch.backends.cudnn as cudnn
from numpy import random
import copy
from models.experimental import attempt_load
from utils.datasets import letterbox
from utils.general import check_img_size, non_max_suppression_face, apply_classifier, scale_coords, xyxy2xywh, \
strip_optimizer, set_logging, increment_path
from utils.plots import plot_one_box
from utils.torch_utils import select_device, load_classifier, time_synchronized
def load_model(weights, device):
model = attempt_load(weights, map_location=device) # load FP32 model
return mo