From 93ae89ece3d9464d66251a95c57680a55d4354a0 Mon Sep 17 00:00:00 2001
From: TG <905865530@qq.com>
Date: Sun, 23 Jun 2024 23:59:57 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E5=A2=9E=E5=8A=A0=E7=99=BE=E9=A6=99?=
 =?UTF-8?q?=E6=9E=9C=E9=87=8D=E9=87=8F=E5=A4=84=E7=90=86=E9=83=A8=E5=88=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 20240529RGBtest3/classifer.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/20240529RGBtest3/classifer.py b/20240529RGBtest3/classifer.py
index f029739..73a18b2 100644
--- a/20240529RGBtest3/classifer.py
+++ b/20240529RGBtest3/classifer.py
@@ -10,6 +10,7 @@ import json
 import utils
 import joblib
 import logging
+import random
 import numpy as np
 from PIL import Image
 from sklearn.ensemble import RandomForestRegressor
@@ -519,7 +520,10 @@ class Data_processing:
         max_mask = pf.find_largest_component(combined_mask)
         contour_mask = self.contour_process(max_mask)
         long_axis, short_axis = self.analyze_ellipse(contour_mask)
+        #重量单位为g，加上了一点随机数
         weight = self.weight_estimates(long_axis, short_axis)
+        weight = (weight * 2) + random.randint(0, 30) - random.randint(0, 30)
+
         number_defects, total_pixels = self.analyze_defect(max_mask)
         edge = pf.draw_contours_on_image(img, contour_mask)
         org_defect = pf.bitwise_and_rgb_with_binary(edge, max_mask)