我正在从事一个机器人项目,涉及到人体检测,我正在使用张量流和预定义的数据集来创建一个训练模型。由于我是机器学习的新手,我无法正确地从分类器中获得输出。我只要求人检测,并想避免检测球,笔记本电脑或其他物体。
现在我的摄像头可以检测到所有的物体,比如球,球棒,笔记本电脑,电视等等。我需要的输出只有80%以上的人。在
我使用创建的模型的代码是import numpy as np
import os
import six.moves.urllib as urllib
import sys
import tarfile
import tensorflow as tf
import zipfile
from collections import defaultdict
from io import StringIO
from matplotlib import pyplot as plt
from PIL import Image
from utils import label_map_util
from utils import visualization_utils as vis_util
MODEL_NAME = 'ssd_mobilenet_v1_coco_11_06_2017'
MODEL_FILE = MODEL_NAME + '.tar.gz'
DOWNLOAD_BASE = 'http://download.tensorflow.org/models/object_detection/'
PATH_TO_CKPT = MODEL_NAME + '/f