我想微调PyTorch中的目标检测器.为此,我使用了本教程:
但是,FastRCNN模型不适合我的用例,所以我对SSDLight进行了微调。我编写这段代码是为了设置一个新的分类头:
from functools import partial
from torchvision.models.detection import _utils as det_utils
from torchvision.models.detection.ssdlite import SSDLiteClassificationHead
model = torchvision.models.detection.
我正在使用tensorflow lite来检测对象,特别是来自google的SSDLite-MobileNet-v2对象检测模型。除了对象检测之外,我还使用我在youtube上提到的本教程中提到的opencv.For对特定对象执行颜色检测。
首先,将帧读取转换为HSV格式,找出目标包围框中心像素的HSV值,并对颜色进行估计。
对于对象检测,我指的是以下GitHub代码--
物体和颜色检测代码:-
class VideoStream:
"""Camera object that controls video streaming from the Picamera