import?numpy?as?np
import?os
import?cv2

class?Dataset:

????def?__init__（self?imagePath?extensions）:
????????self.data?=?createImageList（imagePath?extensions）
????????np.random.shuffle（self.data）
????????self.num_records?=?len（self.data）
????????self.next_record?=?0

????????self.labels?self.inputs?=?zip（*self.data）

????????category?=?np.unique（self.labels）
????????self.num_labels?=?len（category）
????????self.category2label?=?dict（zip（category?range（len（category））））
????????self.label2category?=?{l:?k?for?k?l?in?self.category2label.items（）}

????????#?Convert?the?labels?to?numbers
????????self.labels?=?[self.category2label[l]?for?l?in?self.labels]

????def?__len__（self）:
????????return?self.num_records

????def?onehot（self?label）:
????????v?=?np.zeros（self.num_labels）
????????v[la

?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件????????4203??2017-12-20?08:55??importData.py
?????文件????????3468??2016-07-07?16:47??inference.py
?????文件?????????720??2016-07-07?16:47??predict_image.py
?????文件????????1088??2017-12-20?09:49??test.py
?????文件????????2589??2017-12-20?09:17??train.py
?????文件?????????170??2016-07-07?16:47??README.md