Spaces:
Runtime error
Runtime error
bills
commited on
Commit
•
11ad137
1
Parent(s):
56f6107
Add confidence and IoU threshold on fine_tune.py
Browse files
__pycache__/multiapp.cpython-37.pyc
CHANGED
|
Binary files a/__pycache__/multiapp.cpython-37.pyc and b/__pycache__/multiapp.cpython-37.pyc differ
|
|
|
apps/__pycache__/fine_tune.cpython-37.pyc
CHANGED
|
Binary files a/apps/__pycache__/fine_tune.cpython-37.pyc and b/apps/__pycache__/fine_tune.cpython-37.pyc differ
|
|
|
apps/__pycache__/home.cpython-37.pyc
CHANGED
|
Binary files a/apps/__pycache__/home.cpython-37.pyc and b/apps/__pycache__/home.cpython-37.pyc differ
|
|
|
apps/__pycache__/main_model.cpython-37.pyc
CHANGED
|
Binary files a/apps/__pycache__/main_model.cpython-37.pyc and b/apps/__pycache__/main_model.cpython-37.pyc differ
|
|
|
apps/fine_tune.py
CHANGED
|
@@ -23,6 +23,8 @@ def app():
|
|
| 23 |
prediction = st.button("Predict")
|
| 24 |
if prediction:
|
| 25 |
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/fine_tune.pt", force_reload=True)
|
|
|
|
|
|
|
| 26 |
results = ship_model(f"{folder_path}")
|
| 27 |
with st.spinner("Loading..."):
|
| 28 |
time.sleep(3.5)
|
|
|
|
| 23 |
prediction = st.button("Predict")
|
| 24 |
if prediction:
|
| 25 |
ship_model = torch.hub.load('ultralytics/yolov5', 'custom', path="apps/model/fine_tune.pt", force_reload=True)
|
| 26 |
+
ship_model.conf = 0.6
|
| 27 |
+
ship_model.iou = 0.55
|
| 28 |
results = ship_model(f"{folder_path}")
|
| 29 |
with st.spinner("Loading..."):
|
| 30 |
time.sleep(3.5)
|