from?__future__?import?print_function
from?PIL?import?Image
import?os
import?sys
import?numpy?as?np
import?argparse

import?torch.utils.data?as?data


class?Fruit（data.Dataset）:
????
????def?__init__（self?root_dir?train=True?transform=None）:
????????self.root_dir?=?os.path.abspath（root_dir）
????????self.transform?=?transform
????????self.train=train

????????if?（self.train）:
????????????self.data?=?np.load（os.path.join（self.root_dir?“train_data.npy“））
????????????self.labels?=?np.load（os.path.join（self.root_dir?“train_labels.npy“））
????????else:
????????????self.data?=?np.load（os.path.join（self.root_dir?“validation_data.npy“））
????????????self.labels?=?np.load（os.path.join（self.root_dir?“validation_labels.npy“））

????????self.data?=?self.data.transpose（（0?2?3?1））
????
????def?__getitem__（self?index）:
????????img?target?=?self.data[index]?self.labels[index]
????????#img?=?Image.fromarray（img.astype（‘uint8‘））
????????
????????if?self.transform?is?not?None:
????????????img?=?self.transform（img）

????????return?img?target

????def?__len__（self）:
????????return?（len（self.data））

?????

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-03-28?23:33??fruit-classification-pytorch-master\
?????文件????????1203??2019-03-28?23:33??fruit-classification-pytorch-master\.gitignore
?????文件????????1084??2019-03-28?23:33??fruit-classification-pytorch-master\LICENSE
?????文件????????1455??2019-03-28?23:33??fruit-classification-pytorch-master\README.md
?????文件????????1111??2019-03-28?23:33??fruit-classification-pytorch-master\fruit_data.py
?????文件????????4090??2019-03-28?23:33??fruit-classification-pytorch-master\load_dataset.py
?????目錄???????????0??2019-03-28?23:33??fruit-classification-pytorch-master\logs\
?????文件?????1277608??2019-03-28?23:33??fruit-classification-pytorch-master\logs\training_log.txt
?????目錄???????????0??2019-03-28?23:33??fruit-classification-pytorch-master\model\
?????文件?????1689894??2019-03-28?23:33??fruit-classification-pytorch-master\model\fruit_model_optimizer_dict.pth
?????文件?????1689770??2019-03-28?23:33??fruit-classification-pytorch-master\model\fruit_model_state_dict.pth
?????目錄???????????0??2019-03-28?23:33??fruit-classification-pytorch-master\test\
?????文件??????266418??2019-03-28?23:33??fruit-classification-pytorch-master\test\experiment.csv
?????文件?????????523??2019-03-28?23:33??fruit-classification-pytorch-master\test\generate_loss_report.py
?????文件???????15895??2019-03-28?23:33??fruit-classification-pytorch-master\test\test_image.png
?????文件????????1652??2019-03-28?23:33??fruit-classification-pytorch-master\test\test_image.py
?????文件???????15895??2019-03-28?23:33??fruit-classification-pytorch-master\test_image.png
?????文件????????4551??2019-03-28?23:33??fruit-classification-pytorch-master\train.py