def temp_import_detector(path="/Users/aub3/tempd"): """ For testing pre-developed detectors :param path: :return: """ setup_django() from dvaapp.shared import create_detector_folders import json from django.conf import settings from dvaapp.models import CustomDetector d = CustomDetector() with open("{}/input.json".format(path)) as infile: data = json.load(infile) d.name = "test detector" d.class_names = json.dumps(data['class_names']) d.phase_1_log = file("{}/phase_1.log".format(path)).read d.phase_2_log = file("{}/phase_2.log".format(path)).read d.frames_count = 500 d.boxes_count = 500 d.class_distribution = json.dumps(data['class_names']) d.save() create_detector_folders(d) shutil.copy("{}/phase_2_best.h5".format(path),"{}/detectors/{}/phase_2_best.h5".format(settings.MEDIA_ROOT,d.pk))
def test_train_yolo(): """ :return: """ setup_django() from dvaapp.models import TEvent, CustomDetector from dvaapp.tasks import train_yolo_detector from django.conf import settings args = {} detector = CustomDetector() detector.save() args['detector_pk'] = detector.pk args['object_names'] = [ "red_buoy", "green_buoy", "yellow_buoy", "path_marker", "start_gate", "channel" ] args['root_dir'] = "{}/models/{}/".format(settings.MEDIA_ROOT, detector.pk) train_yolo_detector( TEvent.objects.create(arguments_json=json.dumps(args)).pk)
def temp_import_detector(path="/Users/aub3/tempd"): """ For testing pre-developed detectors :param path: :return: """ setup_django() from dvaapp.shared import create_detector_folders import json from django.conf import settings from dvaapp.models import CustomDetector d = CustomDetector() with open("{}/input.json".format(path)) as infile: data = json.load(infile) d.name = "test detector" d.class_names = json.dumps(data['class_names']) d.phase_1_log = file("{}/phase_1.log".format(path)).read d.phase_2_log = file("{}/phase_2.log".format(path)).read d.frames_count = 500 d.boxes_count = 500 d.class_distribution = json.dumps(data['class_names']) d.save() create_detector_folders(d) shutil.copy("{}/phase_2_best.h5".format(path),"{}/detectors/{}/phase_2_best.h5".format(settings.MEDIA_ROOT,d.pk))