Zaktualizuj 'Find_animal.py'
This commit is contained in:
parent
ec6ee7f2fa
commit
dac576cc44
@ -80,7 +80,7 @@ def postprocess(boxes, masks):
|
||||
|
||||
|
||||
# załaduj nazwy
|
||||
classesFile = "mscoco_labels.names";
|
||||
classesFile = "labels.names";
|
||||
classes = None
|
||||
with open(classesFile, 'rt') as f:
|
||||
classes = f.read().rstrip('\n').split('\n')
|
||||
@ -114,14 +114,14 @@ if (args.image):
|
||||
print("Input image file ", args.image, " doesn't exist")
|
||||
sys.exit(1)
|
||||
cap = cv.VideoCapture(args.image)
|
||||
outputFile = args.image[:-4]+'_mask_rcnn_out_py.jpg'
|
||||
outputFile = args.image[:-4]+'found_animal.jpg'
|
||||
elif (args.video):
|
||||
# Otwiera plik video
|
||||
if not os.path.isfile(args.video):
|
||||
print("Input video file ", args.video, " doesn't exist")
|
||||
sys.exit(1)
|
||||
cap = cv.VideoCapture(args.video)
|
||||
outputFile = args.video[:-4]+'_mask_rcnn_out_py.avi'
|
||||
outputFile = args.video[:-4]+'found_animal.avi'
|
||||
|
||||
|
||||
# inicjalizacja zapisu video
|
||||
@ -134,8 +134,8 @@ while cv.waitKey(1) < 0:
|
||||
|
||||
# zamyka program, gdy skończy się film
|
||||
if not hasFrame:
|
||||
print("Done processing !!!")
|
||||
print("Output file is stored as ", outputFile)
|
||||
print("Processing Done.")
|
||||
print("Output file saved ad: ", outputFile)
|
||||
cv.waitKey(3000)
|
||||
break
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user