Spaces:
Sleeping
Sleeping
File size: 2,690 Bytes
bf4a716 0c9b662 52dcbd6 0c9b662 437c602 0c9b662 bf4a716 0c9b662 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import streamlit as st
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import os
import tempfile
import shutil
# Coba Import YOLO
try:
from ultralytics import YOLO
YOLO_AVAILABLE = True
except ImportError:
YOLO_AVAILABLE = False
st.set_page_config(page_title="Mineral Classification")
# Periksa apakah library YOLO tersedia
def cek_library():
if not YOLO_AVAILABLE:
st.error("Ultralytics tidak terpasang. Silakan instal dengan perintah berikut:")
st.code("pip install ultralytics")
return False
return True
st.markdown("""
<div style="background-color:#0984e3; padding: 20px; text-align: center;">
<h1 style="color: white;"> Mineral Classifier Program </h1>
<h5 style="color: white;"> Mineral Image Detection </h5>
</div>
""", unsafe_allow_html=True)
# Pastikan library sudah terpasang sebelum melanjutkan
if cek_library():
uploaded_file = st.file_uploader("Upload Gambar Mineral", type=['jpg', 'jpeg', 'png'])
if uploaded_file:
temp_dir = tempfile.mkdtemp()
temp_file = os.path.join(temp_dir, "gambar.jpg")
image = Image.open(uploaded_file)
# Ubah Ukuran Gambar
image = image.resize((300, 300))
image.save(temp_file)
# Tampilkan gambar
st.markdown("<div style='text-align: center;'>", unsafe_allow_html=True)
st.image(image, caption="Gambar yang diupload")
st.markdown("</div>", unsafe_allow_html=True)
# Deteksi Gambar
if st.button("Deteksi Gambar"):
with st.spinner("Sedang diproses"):
try:
model = YOLO('besst.pt') # nama model kamu
hasil = model(temp_file)
nama_objek = hasil[0].names
nilai_prediksi = hasil[0].probs.data.numpy().tolist()
objek_terdeteksi = nama_objek[np.argmax(nilai_prediksi)]
fig, ax = plt.subplots()
ax.bar(list(nama_objek.values()), nilai_prediksi)
ax.set_title('Tingkat Keyakinan Prediksi')
ax.set_xlabel('Mineral')
ax.set_ylabel('Keyakinan')
plt.xticks(rotation=45)
st.success(f"Mineral detected: {objek_terdeteksi}")
st.pyplot(fig)
except Exception as e:
st.error("Gambar tidak dapat terdeteksi")
st.error(f"Error: {e}")
shutil.rmtree(temp_dir, ignore_errors=True)
st.markdown(
"<div style='text-align: center;' class='footer'> Mineral Detection Application Program </div>",
unsafe_allow_html=True
) |