首页 Paddle Lite 帖子详情
训练自定义数据集时,预测没有产生结果 已解决
收藏
快速回复
Paddle Lite 问答推理部署 1016 7
训练自定义数据集时,预测没有产生结果 已解决
收藏
快速回复
Paddle Lite 问答推理部署 1016 7

执行预测命令:python -u tools/infer.py -c configs/yolov3_mobilenet_v1_voc.yml --infer_img=dataset/1.jpeg

输出的图片没有产生bbox,lable名,图片生成了,跟原始的一样,没有产生结果,是哪里有问题?参数设置有问题吗?

输出结果:2021-09-23 10:42:49,393 - INFO - Not found annotation file dataset\label_list.tx
t, load voc2012 categories.
2021-09-23 10:42:50,510 - INFO - Infer iter 0
2021-09-23 10:42:50,543 - INFO - Detection bbox results save in output\1.jpeg

 

 

yml文件如下:(lable_list一共有两类)

architecture: YOLOv3
use_gpu: false
max_iters: 168
log_iter: 20
save_dir: output
snapshot_iter: 2000
metric: VOC
map_type: 11point
pretrain_weights: https://paddlemodels.bj.bcebos.com/object_detection/yolov3_mobilenet_v1.tar
weights: output/yolov3_mobilenet_v1_voc/model_final
num_classes: 2
use_fine_grained_loss: false

YOLOv3:
backbone: MobileNet
yolo_head: YOLOv3Head

MobileNet:
norm_type: sync_bn
norm_decay: 0.
conv_group_scale: 1
with_extra_blocks: false

YOLOv3Head:
anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]]
anchors: [[10, 13], [16, 30], [33, 23],
[30, 61], [62, 45], [59, 119],
[116, 90], [156, 198], [373, 326]]
norm_decay: 0.
yolo_loss: YOLOv3Loss
nms:
background_label: -1
keep_top_k: 100
nms_threshold: 0.45
nms_top_k: 1000
normalized: false
score_threshold: 0.01

YOLOv3Loss:
ignore_thresh: 0.7
label_smooth: false

LearningRate:
base_lr: 0.001
schedulers:
- !PiecewiseDecay
gamma: 0.1
milestones:
- 55000
- 62000
- !LinearWarmup
start_factor: 0.
steps: 1000

OptimizerBuilder:
optimizer:
momentum: 0.9
type: Momentum
regularizer:
factor: 0.0005
type: L2

_READER_: 'yolov3_reader.yml'
TrainReader:
dataset:
!VOCDataSet
dataset_dir: dataset
anno_path: trainval.txt
use_default_label: false
batch_size: 2

EvalReader:
inputs_def:
fields: ['image', 'im_size', 'im_id', 'gt_bbox', 'gt_class', 'is_difficult']
dataset:
!VOCDataSet
dataset_dir: dataset
anno_path: test.txt
use_default_label: false

TestReader:
dataset:
!ImageFolder
dataset_dir: dataset
anno_path: label_list.txt
use_default_label: false

yxxyuyi
已解决
8# 回复于2021-10
python -u tools/infer.py -c configs/yolov3_mobilenet_v1_voc.yml --infer_img=dataset/aa.mp4 --draw_threshold=0.2 draw_threshold修改此参数可正常识别出来
展开
0
收藏
回复
全部评论(7)
时间顺序
FutureSI
#2 回复于2021-09

报错没有找到类别列表 label_list.txt

0
回复
八宝粥
#3 回复于2021-09

2021-09-23 10:42:49,393 - INFO - Not found annotation file dataset\label_list.tx

你这错误不是都报出来了吗?

0
回复
yxxyuyi
#4 回复于2021-10
2021-09-23 10:42:49,393 - INFO - Not found annotation file dataset\label_list.tx 你这错误不是都报出来了吗?

D:\paddle\PaddleDetection-release-2.2\PaddleDetection-release-2.2\static>python -u tools/infer.py -c configs/yolov3_mobilenet_v1_voc.yml --infer_img=dataset/1.jpeg
2021-10-08 13:11:25,100-INFO: Load categories from dataset\label_list.txt
2021-10-08 13:11:26,016-INFO: Infer iter 0
2021-10-08 13:11:26,049-INFO: Detection bbox results save in output\1.jpeg

0
回复
yxxyuyi
#5 回复于2021-10
报错没有找到类别列表 label_list.txt

D:\paddle\PaddleDetection-release-2.2\PaddleDetection-release-2.2\static>python -u tools/infer.py -c configs/yolov3_mobilenet_v1_voc.yml --infer_img=dataset/1.jpeg
2021-10-08 13:11:25,100-INFO: Load categories from dataset\label_list.txt
2021-10-08 13:11:26,016-INFO: Infer iter 0
2021-10-08 13:11:26,049-INFO: Detection bbox results save in output\1.jpeg

0
回复
FutureSI
#6 回复于2021-10

Load categories from dataset\label_list.txt 意思是正在loading,不是loaded吧

0
回复
FutureSI
#7 回复于2021-10

不行照着demo挨个换数据集的部分,就能看出哪部分毛病了。

0
回复
yxxyuyi
#8 回复于2021-10

python -u tools/infer.py -c configs/yolov3_mobilenet_v1_voc.yml --infer_img=dataset/aa.mp4 --draw_threshold=0.2

draw_threshold修改此参数可正常识别出来

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户