未验证 提交 27bc5e0a 编写于 作者: W wangguanzhong 提交者: GitHub

fix fruit dataset (#3806)

* fix fruit dataset

* fix md5
上级 e0076bb2
......@@ -63,8 +63,8 @@ OptimizerBuilder:
YoloTrainFeed:
batch_size: 1
dataset:
dataset_dir: dataset/fruit/fruit-detection
annotation: train.txt
dataset_dir: dataset/fruit
annotation: fruit-detection/train.txt
use_default_label: false
num_workers: 16
bufsize: 128
......@@ -109,8 +109,8 @@ YoloEvalFeed:
batch_size: 1
image_shape: [3, 608, 608]
dataset:
dataset_dir: dataset/fruit/fruit-detection
annotation: val.txt
dataset_dir: dataset/fruit
annotation: fruit-detection/val.txt
use_default_label: false
......@@ -118,5 +118,5 @@ YoloTestFeed:
batch_size: 1
image_shape: [3, 608, 608]
dataset:
dataset_dir: dataset/fruit/fruit-detection
dataset_dir: dataset/fruit
use_default_label: false
......@@ -71,11 +71,9 @@ DATASETS = {
'https://dataset.bj.bcebos.com/wider_face/wider_face_split.zip',
'a4a898d6193db4b9ef3260a68bad0dc7', ),
], ["WIDER_train", "WIDER_val", "wider_face_split"]),
'fruit': ([
(
'https://dataset.bj.bcebos.com/PaddleDetection_demo/fruit-detection.tar',
'374554a7633b1b68d6a5fbb7c061b8ba', ),
], ["fruit-detection"]),
'fruit': ([(
'https://dataset.bj.bcebos.com/PaddleDetection_demo/fruit-detection.tar',
'ee4a1bf2e321b75b0850cc6e063f79d7', ), ], ["fruit-detection"]),
}
DOWNLOAD_RETRY_LIMIT = 3
......@@ -133,8 +131,8 @@ def get_dataset_path(path, annotation, image_dir):
# not match any dataset in DATASETS
raise ValueError("Dataset {} is not valid and cannot parse dataset type "
"'{}' for automaticly downloading, which only supports "
"'voc' and 'coco' currently".format(path,
osp.split(path)[-1]))
"'voc', 'coco', 'wider_face' and 'fruit' currently".format(
path, osp.split(path)[-1]))
def create_voc_list(data_dir, devkit_subdir='VOCdevkit'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册