yolov5-lite
NanoDet-Plus
YoloX
WIDER FACE是一个人脸数据集合，其中包含了人脸的标注，可以用它来转化为yolo的标注格式，做训练，极大节约成本
转化代码：
import os
def convert_wider_to_yolo(annotation_file, image_folder, output_folder):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
with ope[......]
思路：使用模型的特征提取层，转化成向量，然后比对向量的距离(比如cos)
import torch
import torchvision.models as models
import torchvision.transforms as transforms
from PIL import Image
# load model & feature only
model = models.mobilenet_v3_small(pretrained=True)
mode[......]
HuggingFace的transformers库提供了各种SOTA开源模型的方案，而且做了整合链，很方便使用。
1 安装
pip install transformers
2 做情感分析任务
数据准备
from transformers import pipeline
import numpy as np
import pandas as pd
import seaborn as sn
# prepare data
df = pd.read_csv("Airline[......]