alibaba / pipcook

Machine learning platform for Web developers
https://alibaba.github.io/pipcook/
Apache License 2.0
2.55k stars 209 forks source link

示例文档是否错误? #753

Closed ShanaMaid closed 3 years ago

ShanaMaid commented 3 years ago

image

下载的passcalvoc数据集,为什么用coco进行分析呢?

rickycao-qy commented 3 years ago

第二层是数据接入层,类似于数据集合模型之间的桥梁,如果下载的是 pascal 的数据集,但是模型需要的是 coco 格式,那么需要 coco-data-access 进行转化

ShanaMaid commented 3 years ago

执行结果和文档描述不一致呢

const predict = require('./output');
(async () => {
  const v1 = await predict('./zz.png');
  console.log(v1); 

})();

控制台输出:

Overriding model.yaml nc=80 with nc=2

[None]

生成的文件: image

hyp-define.yaml

lr0: 0.01
lrf: 0.2
momentum: 0.937
weight_decay: 0.0005
warmup_epochs: 3.0
warmup_momentum: 0.8
warmup_bias_lr: 0.1
box: 0.05
cls: 0.5
cls_pw: 1.0
obj: 1.0
obj_pw: 1.0
iou_t: 0.2
anchor_t: 4.0
fl_gamma: 0.0
hsv_h: 0.015
hsv_s: 0.7
hsv_v: 0.4
degrees: 0.0
translate: 0.1
scale: 0.5
shear: 0.0
perspective: 0.0
flipud: 0.0
fliplr: 0.5
mosaic: 1.0
mixup: 0.0

opt-define.yaml

hyp: ./output/node_modules/@pipcook/plugins-pytorch-yolov5-model-define/config/hyp.scratch.yaml
device: ''
total_batch_size: 16
batch_size: 16
nc: 2
weights: yolov5s.pt
cfg: yolov5s.yaml
img_size:
- 640
- 640

文档示例: image

rickycao-qy commented 3 years ago

是完全执行的原始的教程的步骤吗,图片也是一样的吗

ShanaMaid commented 3 years ago

用label-image自己标注的图片,为了快速测试,只配置了一张。

rickycao-qy commented 3 years ago

应该是因为当前图片没有检测出目标,所以会返回 None

ShanaMaid commented 3 years ago

识别图片和训练图片用的都是同一张

ShanaMaid commented 3 years ago

并且直接require('./output')会报错

import sys
^^^^^^

SyntaxError: Cannot use import statement outside a module
    at wrapSafe (internal/modules/cjs/loader.js:915:16)
    at Module._compile (internal/modules/cjs/loader.js:963:27)
    at Object.Module._extensions..js (internal/modules/cjs/loader.js:1027:10)
    at Module.load (internal/modules/cjs/loader.js:863:32)
    at Function.Module._load (internal/modules/cjs/loader.js:708:14)

参照https://github.com/alibaba/pipcook/issues/690#issuecomment-759451928修改了_requirePlugin才能执行。

执行结果就是

Overriding model.yaml nc=80 with nc=2

[None]
rickycao-qy commented 3 years ago

是不是预测逻辑有点问题,模型定义插件的源码在 https://github.com/imgcook/pipcook-plugin-pytorch-yolov5-model/blob/main/__init__.py 我们尽快检查一下,同时有兴趣也可以贡献一下哦,python 插件预测的时候加载报错的问题会同步支持一下

yorkie commented 3 years ago

@rickycao-qy @FeelyChau 我们是不是可以在 2.0 版本中支持目标检测?

FeelyChau commented 3 years ago

@rickycao-qy @FeelyChau 我们是不是可以在 2.0 版本中支持目标检测?

正在考虑如何迁移 YOLO v5.

FeelyChau commented 3 years ago

@ShanaMaid Pipcook 2.0 已支持目标检测,参考文档: https://alibaba.github.io/pipcook/#/zh-cn/