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

fix fruit dataset (#3806)

* fix fruit dataset

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