Closed ShanaMaid closed 3 years ago
第二层是数据接入层,类似于数据集合模型之间的桥梁,如果下载的是 pascal 的数据集,但是模型需要的是 coco 格式,那么需要 coco-data-access 进行转化
执行结果和文档描述不一致呢
const predict = require('./output');
(async () => {
const v1 = await predict('./zz.png');
console.log(v1);
})();
控制台输出:
Overriding model.yaml nc=80 with nc=2
[None]
生成的文件:
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
文档示例:
是完全执行的原始的教程的步骤吗,图片也是一样的吗
用label-image自己标注的图片,为了快速测试,只配置了一张。
应该是因为当前图片没有检测出目标,所以会返回 None
识别图片和训练图片用的都是同一张
并且直接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]
是不是预测逻辑有点问题,模型定义插件的源码在 https://github.com/imgcook/pipcook-plugin-pytorch-yolov5-model/blob/main/__init__.py 我们尽快检查一下,同时有兴趣也可以贡献一下哦,python 插件预测的时候加载报错的问题会同步支持一下
@rickycao-qy @FeelyChau 我们是不是可以在 2.0 版本中支持目标检测?
@rickycao-qy @FeelyChau 我们是不是可以在 2.0 版本中支持目标检测?
正在考虑如何迁移 YOLO v5.
@ShanaMaid Pipcook 2.0 已支持目标检测,参考文档: https://alibaba.github.io/pipcook/#/zh-cn/
下载的passcalvoc数据集,为什么用coco进行分析呢?