Commit 8b491311 authored by Ubuntu's avatar Ubuntu
Browse files

Fixed conditional use of AIcrowd API, commented debug print

parent 09dfbc2f
...@@ -168,7 +168,7 @@ def predict_output (image): ...@@ -168,7 +168,7 @@ def predict_output (image):
t1 = time.time() t1 = time.time()
print("PREDICTIONS, DONE IN", round(t1 - t0, 3), "s:") print("PREDICTIONS, DONE IN", round(t1 - t0, 3), "s:")
classes = [p['class_ids'] for p in predictions] classes = [p['class_ids'] for p in predictions]
print(classes) # print(classes)
p = predictions[0] p = predictions[0]
# Subtract 1 from each class ID (0 is background) # Subtract 1 from each class ID (0 is background)
......
...@@ -6,8 +6,10 @@ import base64 ...@@ -6,8 +6,10 @@ import base64
import glob import glob
import os import os
import traceback import traceback
import platform
from PIL import Image from PIL import Image
use_aicrowd_helpers = False
use_aicrowd_helpers = platform.platform() != "Linux-4.4.0-1095-aws-x86_64-with-debian-stretch-sid"
if use_aicrowd_helpers: if use_aicrowd_helpers:
import aicrowd_helpers import aicrowd_helpers
# from eval import evaluate # from eval import evaluate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment