Python DetectionModelTrainer.setGpuUsage примеры использования

Пример #1

Файл: trainer.py Проект: Kaustik/icanhazcoffee

from imageai.Detection.Custom import DetectionModelTrainer
import os

execution_path = os.getcwd()

trainer = DetectionModelTrainer()
trainer.setGpuUsage(1)
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory=".")
trainer.setTrainConfig(object_names_array=["empty coffee", "not empty coffee"],
                       batch_size=4,
                       num_experiments=10,
                       train_from_pretrained_model="pretrained-yolov3.h5")
trainer.trainModel()

Пример #2

Файл: train.py Проект: Emilien-mipt/HackerEarth-Cartoon_emotion_detection

from imageai.Detection.Custom import DetectionModelTrainer

trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory="../data/dataset/")
trainer.setGpuUsage("0")
trainer.setTrainConfig(
    object_names_array=["Tom", "Jerry"],
    batch_size=2,
    num_experiments=100,
    train_from_pretrained_model="../models/pretrained-yolov3.h5",
)
trainer.trainModel()