导出到Tensorflow.js模型的自定义AutoML Edge训练对象检测模型返回高度错误的边界框

问题描述

我已经训练了一个自定义的AutoML Edge对象检测模型,并将其导出到Tensorflowjs模型。当使用get_parameter演示程序在浏览器中运行且更新了相关性时:

            Criteria criteria=new Criteria();
            Query query= new Query();
            if(ownerId!=null && ownerId>0) {
                query.addCriteria(criteria.andOperator(Criteria.where("OwnerId").is(ownerId)));
            }
            if(env!=null) {
                query.addCriteria(criteria.andOperator(Criteria.where("Environment").is(env)));
            }
            if(api!=null) {
                query.addCriteria(criteria.andOperator(Criteria.where("ServiceName").in(api)));
            }
            MatchOperation matchOp= Aggregation.match("How to set the query here?");
            Projectionoperation projOp= Aggregation.project("SubServices").andInclude("ServerGroupName").andInclude("NodeName").andInclude("ServiceVersion")
                    .andExclude("_id");
            Aggregation aggr = Aggregation.newAggregation(matchOp,projOp);
            AggregationResults<Document> aggregate = mongoOperations.aggregate(aggr,"CNF_SERVICE",Document.class,"config",env);

我注意到输出边界框的高度不正确。

我还可以使用提供的演示模型和图像来触发问题:

https://github.com/tensorflow/tfjs/tree/master/tfjs-automl/demo/object_detection

在运行具有原始演示提供的依赖关系的演示时:

"dependencies": {
    "@tensorflow/tfjs-converter": "^2.0.1","@tensorflow/tfjs-core": "^2.0.1","@tensorflow/tfjs-backend-webgl": "^2.0.1","@tensorflow/tfjs-automl": "^1.0.0"
  }

我得到以下输出

Image and detection output using  dependencies above

有人可以帮助我弄清楚如何解决此问题吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)