def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python_' + __name__
# module_name = 'python:camtrawl.processes'
# module_name = 'python' + __name__
if process_factory.is_process_module_loaded(module_name):
return
# print('TMP_SPROKIT_PROCESS_REGISTRY = {}'.format(ub.repr2(TMP_SPROKIT_PROCESS_REGISTRY)))
for name, doc, cls in TMP_SPROKIT_PROCESS_REGISTRY:
# print("REGISTER PROCESS:")
# print(' * name = {!r}'.format(name))
# print(' * cls = {!r}'.format(cls))
process_factory.add_process(name, doc, cls)
# process_factory.add_process('camtrawl_detect_fish',
# 'preliminatry detection / feature extraction',
# CamtrawlDetectFishProcess)
# process_factory.add_process('camtrawl_measure',
# 'preliminatry measurement',
# CamtrawlMeasureProcess)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:smqtk.smqtk_processes'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process(
'smqtk_ingest_descriptors',
'Add descriptors and parallel UUIDs to a SMQTK descriptor index',
smqtk_ingest_descriptors.SmqtkIngestDescriptors
)
process_factory.add_process(
'smqtk_process_query',
'Perform queries against some arbitrary descriptor index',
smqtk_process_query.SmqtkProcessQuery
)
process_factory.add_process(
'smqtk_object_track_descriptors',
'Add descriptors to object tracks',
smqtk_object_track_descriptors.SmqtkObjectTrackDescriptors
)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.ApplyDescriptor'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('ApplyDescriptor', 'Apply descriptor to image', ApplyDescriptor)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.python.SimpleHomogTracker'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process(
'simple_homog_tracker',
'Simple IOU-based tracker with homography support',
SimpleHomogTracker,
)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:viame.pytorch.MDNetTrackerProcess'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('mdnet_tracker', 'Object tracking using mdnet',
MDNetTrackerProcess)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.pytorch.resnet_descriptors'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('resnet_descriptors',
'resnet feature extraction', ResnetDescriptors)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.ProcessImage'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('ProcessImage', 'Process image test',
ProcessImage)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.pytorch.resnet_augmentation'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('resnet_augmentation',
'Pytorch-Based Augmentation', DataAugmentation)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:test.pythonpath.process_test'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('pythonpath_test_process', 'A test process.',
TestPythonProcess)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.pytorch.SRNNTracker'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('srnn_tracker',
'Structural RNN based tracking',
SRNNTracker)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.print_number'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('kw_print_number_process',
'A Simple Kwiver Test Process',
kw_print_number_process)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.write_homography'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('kw_write_homography',
'A Simple Kwiver homography writer',
HomographyWriterProcess)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:hello_world.hello_world_detector'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('hello_world_detector', 'Example detector',
hello_world_detector.hello_world_detector)
process_factory.add_process('hello_world_filter', 'Example filter',
hello_world_filter.hello_world_filter)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:kwiver.sprokit.tests.processes'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process('test_python_process', 'A test Python process',
TestPythonProcess)
process_factory.add_process('pyprint_number',
'A Python process which prints numbers',
PythonPrintNumberProcess)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:opencv.opencv_processes'
if process_factory.is_process_module_loaded(module_name):
return
process_factory.add_process(
'ocv_multimodal_registration', 'Register optical and thermal frames',
ocv_multimodal_registration.register_frames_process)
process_factory.add_process(
'ocv_fft_filter_based_on_ref',
'Filter image in the frequency based on some template',
ocv_fft_filter_based_on_ref.filter_based_on_ref_process)
process_factory.mark_process_module_as_loaded(module_name)
def __sprokit_register__():
from kwiver.sprokit.pipeline import process_factory
module_name = 'python:core.core_processes'
if process_factory.is_process_module_loaded( module_name ):
return
process_factory.add_process(
'blank_out_frames',
'Blank out frames with no object detections on them',
utility_processes.blank_out_frames
)
process_factory.add_process(
'percentile_norm_npy_16_to_8bit',
'A specialized percentile normalization method',
utility_processes.percentile_norm_npy_16_to_8bit
)
process_factory.mark_process_module_as_loaded( module_name )
