加载图片 Parameters: dirName - 文件夹的名字 Returns: trainingMat - 数据矩阵 hwLabels - 数据标签
(dirName)
| 271 | return returnVect |
| 272 | |
| 273 | def loadImages(dirName): |
| 274 | """ |
| 275 | 加载图片 |
| 276 | Parameters: |
| 277 | dirName - 文件夹的名字 |
| 278 | Returns: |
| 279 | trainingMat - 数据矩阵 |
| 280 | hwLabels - 数据标签 |
| 281 | """ |
| 282 | from os import listdir |
| 283 | hwLabels = [] |
| 284 | trainingFileList = listdir(dirName) |
| 285 | m = len(trainingFileList) |
| 286 | trainingMat = np.zeros((m,1024)) |
| 287 | for i in range(m): |
| 288 | fileNameStr = trainingFileList[i] |
| 289 | fileStr = fileNameStr.split('.')[0] |
| 290 | classNumStr = int(fileStr.split('_')[0]) |
| 291 | if classNumStr == 9: hwLabels.append(-1) |
| 292 | else: hwLabels.append(1) |
| 293 | trainingMat[i,:] = img2vector('%s/%s' % (dirName, fileNameStr)) |
| 294 | return trainingMat, hwLabels |
| 295 | |
| 296 | def testDigits(kTup=('rbf', 10)): |
| 297 | """ |
no test coverage detected