Skip to main content
Join the official 2019 Python Developers SurveyStart the survey!

CV Tools related to Object Detection

Project description

Synapse Computer Vision Tools

More details are provided in Notion page.

Requirements

  • Python 3.5+
  • OpenCV
  • Pillow
  • LXML
  • For TF Records - TensorFlow

Installation

Using PIP

pip3 import syncvtools

Visualize predictions from prod

from syncvtools.utils.draw_detections import DrawDetections
from syncvtools.utils.parsers import ProdDetections, TFRecords, TFObjDetAPIDetections

prod_dets = ProdDetections.parse_prod_detections(img_dir='IMAGE_DIRECTORY', predictions_dir='DETECTIONS_FILE_DIR')
drawer = DrawDetections(bbox_line_height=1, threshold=0.5)
for pro_det in prod_dets:
    vis_img = drawer.draw_imageleveldetections(img_dets=prod_dets[pro_det])
    cv2.imshow("predictions", vis_img)
    cv2.waitKey(0)

Visualize predictions from TF training

from syncvtools.utils.draw_detections import DrawDetections
from syncvtools.utils.parsers import ProdDetections, TFRecords, TFObjDetAPIDetections

tf_inf = TFObjDetAPIDetections.parse_detections(detection_file='detections_and_losses.json')
tf_gts = TFRecords.parse(tfrecord_src='val.tfrecord')
#adding image info to predictions/gt
tf_inf += tf_gts
if label_map is not None:
    tf_inf.process_labelmap('path_to_label_map.pbtxt')

drawer = DrawDetections(bbox_line_height=1, threshold=0.5)


for pro_det in tf_inf:
    if not tf_inf[pro_det].detections and not tf_inf[pro_det].ground_truth:
        continue #no gt/detections here
    vis_img = drawer.draw_imageleveldetections(img_dets=tf_inf[pro_det])
    cv2.imshow("predictions/gt from TF inference", vis_img)
    cv2.waitKey(0)

utils

from syncvtools.utils import file_tools as ft

ft.get_file_list_by_ext(dir='input_path', ext=('jpg','png'))

Returns a list of string with full path to files with ext extension.

Project details


Release history Release notifications

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for syncvtools, version 0.1.6
Filename, size File type Python version Upload date Hashes
Filename, size syncvtools-0.1.6-py3-none-any.whl (117.0 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size syncvtools-0.1.6.tar.gz (108.8 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page