diff --git a/dygraph/paddleseg/datasets/rice.py b/dygraph/paddleseg/datasets/rice.py deleted file mode 100644 index f8041526fa2e265e0eac70709e9c295e860df9ad..0000000000000000000000000000000000000000 --- a/dygraph/paddleseg/datasets/rice.py +++ /dev/null @@ -1,56 +0,0 @@ -# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -import os - -from .dataset import Dataset - - -class Rice(Dataset): - def __init__(self, transforms=None, mode='train', download=True): - self.data_dir = "/mnt/liuyi22/PaddlePaddle/POC/rice_dataset" - self.transforms = transforms - self.file_list = list() - self.mode = mode - self.num_classes = 2 - - if mode.lower() not in ['train', 'eval', 'test']: - raise Exception( - "mode should be 'train', 'eval' or 'test', but got {}.".format( - mode)) - - if self.transforms is None: - raise Exception("transform is necessary, but it is None.") - - if mode == 'train': - file_list = os.path.join(self.data_dir, 'train_list.txt') - elif mode == 'eval': - file_list = os.path.join(self.data_dir, 'val_list.txt') - else: - file_list = os.path.join(self.data_dir, 'test_list.txt') - - with open(file_list, 'r') as f: - for line in f: - items = line.strip().split() - if len(items) != 2: - if mode == 'train' or mode == 'eval': - raise Exception( - "File list format incorrect! It should be" - " image_name label_name\\n") - image_path = os.path.join(self.data_dir, items[0]) - grt_path = None - else: - image_path = os.path.join(self.data_dir, items[0]) - grt_path = os.path.join(self.data_dir, items[1]) - self.file_list.append([image_path, grt_path])