atulyakumar97/fire-and-gun-detection

Fire and Gun detection using yolov3 in videos as well as images. Training code, dataset and trained weight file available.

fire-and-gun-detection's Issues

The yolov3.weights file is no longer available

Please provide a download link. Thank you!

How to improve the detect speed?

Hi:
I noticed in your sample videos, there is no jitters or lags at all. But when i run your default code, it has a lot of jitters/lags. A 5 seconds video may need 15 seconds to finish. Looks like the net.forward part is taking too long. How to modify the code in order to get the same effect as you showed in your video? Many thanks!

def detect_objects(img, net, outputLayers):			
	blob = cv2.dnn.blobFromImage(img, scalefactor=0.00392, size=(320, 320), mean=(0, 0, 0), swapRB=True, crop=False)
	net.setInput(blob)
	outputs = net.forward(outputLayers) #this part is taking so long
	return blob, outputs

Error

Great work you did there :)

I am getting an error while running this code, can you please help me.

Opening videos/fire1.mp4 ....
Traceback (most recent call last):
File "yolo.py", line 143, in
start_video(video_path)
File "yolo.py", line 116, in start_video
model, classes, colors, output_layers = load_yolo()
File "yolo.py", line 24, in load_yolo
output_layers = [layers_names[i[0]-1] for i in net.getUnconnectedOutLayers()]
File "yolo.py", line 24, in
output_layers = [layers_names[i[0]-1] for i in net.getUnconnectedOutLayers()]
IndexError: invalid index to scalar variable.

I would be really glad if you could help me in this.
Thank you :)

Your weight and config file can't support other type of code

Your weight and config file can't support other type of code.only support your code.