|
import gradio as gr |
|
import os |
|
|
|
os.system('git clone https://github.com/WongKinYiu/yolov7.git') |
|
|
|
|
|
def detect(inp): |
|
os.system('python ./yolov7/detect.py --weights ./yolov7/runs/train/yolov7/weights/best.pt --conf 0.25 --img-size 640 --source f{inp} "--project","./yolov7/runs/detect ') |
|
otp=inp.split('/')[2] |
|
return f"./yolov7/runs/detect/exp/{otp}" |
|
|
|
|
|
|
|
inp = gr.inputs.Image(type="filepath", label="Input") |
|
output = gr.outputs.Image(type="filepath", label="Output") |
|
|
|
|
|
io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='Pot Hole Detection With Custom YOLOv7 ',) |
|
io.launch(debug=True,share=False) |
|
|
|
|
