NanjingForestryUniversity/supermachine-wood
3
0
Fork 0
mirror of https://github.com/NanjingForestryUniversity/supermachine-wood.git synced 2025-11-08 10:13:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

添加了直方图选点功能（未测试）

Browse Source
This commit is contained in:
FEIJINTI 2023-03-14 11:12:08 +08:00
parent 442ee49b0e
commit aee4cf9715
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
classifer.py
View File

@ -277,6 +277,8 @@ class WoodClass(object):
x = np.concatenate((x, x_hsv), axis=2) x = np.concatenate((x, x_hsv), axis=2)
x = np.reshape(x, (x.shape[0]*x.shape[1], x.shape[2])) x = np.reshape(x, (x.shape[0]*x.shape[1], x.shape[2]))
hist, bins = np.histogram(x[:, 0], bins=10) hist, bins = np.histogram(x[:, 0], bins=10)
hist = hist[1:]
bins = bins[1:]
# x = x[np.argsort(x[:, 0])] # x = x[np.argsort(x[:, 0])]
# x = x[-self.k:, :] # x = x[-self.k:, :]
hist_number = np.argmax(hist) hist_number = np.argmax(hist)