riciii7 commited on
Commit
aec0b85
·
verified ·
1 Parent(s): 1bd1e60
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -92,7 +92,7 @@ async def inference_volume(file: UploadFile):
92
  results = model.predict(
93
  source=img,
94
  conf=0.5,
95
- iou=0.45
96
  )
97
 
98
  detections = []
@@ -109,7 +109,8 @@ async def inference_volume(file: UploadFile):
109
 
110
  # Hitung volume
111
  volume_mm3 = calculate_sphere_volume(width, height)
112
-
 
113
  label = int(box.cls[0].item())
114
  label_name = model.names[label]
115
 
 
92
  results = model.predict(
93
  source=img,
94
  conf=0.5,
95
+ iou=0.2
96
  )
97
 
98
  detections = []
 
109
 
110
  # Hitung volume
111
  volume_mm3 = calculate_sphere_volume(width, height)
112
+ print("VOLUME")
113
+ print(volume_mm3)
114
  label = int(box.cls[0].item())
115
  label_name = model.names[label]
116