final

Files changed (12) hide show

.gitattributes +1 -33
app.py +21 -0
requirements.txt +3 -0
resource/bangarus.jpg +3 -0
resource/bangarus_wiki.jpg +3 -0
resource/best.pt +3 -0
resource/bukata.jpg +3 -0
resource/cobra_wiki.jpg +3 -0
resource/green.jpg +3 -0
resource/naga.jpg +3 -0
resource/najakouthia.jpg +3 -0
resource/najanaja.jpg +3 -0

.gitattributes CHANGED Viewed

@@ -1,34 +1,2 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text























1	*.pt filter=lfs diff=lfs merge=lfs -text
2	+ *.jpg filter=lfs diff=lfs merge=lfs -text

app.py ADDED Viewed

	@@ -0,0 +1,21 @@

+from ultralytics import YOLO
+import streamlit as st
+from PIL import Image
+model = YOLO('resource/best.pt')
+classes = ['Acrochordus-granulatus', 'Ahaetulla-nasuta', 'Ahaetulla-prasina', 'Argyrogena-fasciolata', 'Boiga-cyanea', 'Boiga-cynodon', 'Boiga-multomaculata', 'Boiga-trigonata', 'Bungarus-caeruleus', 'Bungarus-fasciatus', 'Cerberus-rynchops', 'Chrysopelea-ornata', 'Coelognathus-radiatus', 'Daboia-russelii', 'Dendrelaphis-cyanochloris', 'Dendrelaphis-pictus', 'Dendrelaphis-tristis', 'Echis-carinatus', 'Enhydris-enhydris', 'Fordonia-leucobalia', 'Fowlea-piscator', 'Gerarda-prevostiana', 'Homalopsis-buccata', 'Hydrophis-platurus', 'Indotyphlops-braminus', 'Laticauda-colubrina', 'Laticauda-laticaudata', 'Lycodon-aulicus', 'Macropisthodon-plumbicolor', 'Malayopython-reticulatus', 'Naja-kaouthia', 'Naja-naja', 'Oligodon-arnensis', 'Oreocryptophis-porphyraceus', 'Ovophis-monticola', 'Psammodynastes-pulverulentus', 'Ptyas-korros', 'Ptyas-mucosa', 'Python-bivittatus', 'Rhabdophis-subminiatus', 'Sinomicrurus-macclellandi', 'Trimeresurus-albolabris', 'Trimeresurus-erythrurus', 'Trimeresurus-popeiorum', 'Xenochrophis-piscator', 'Nil-object']
+col1, col2 = st.columns([2,1])
+col1.title('Snake Detection System')
+photo = col2.file_uploader('Upload an Image (JPEG)', type='jpg')
+if photo:
+    col2.success('Successfully Uploaded')
+    col1.image(photo)
+    image = Image.open(photo)
+    try:
+        bx = model(image)[0].boxes[0]
+        col1.write(f"Result: {(classes[int(bx.cls[0])])}")
+    except:
+        col1.write(f"Result: Could not detect")