androuino

Caused by: java.lang.RuntimeException: Failed to allocate 27688960 bytes from DEVICE [0] memory
at org.nd4j.jita.memory.CudaMemoryManager.allocate(CudaMemoryManager.java:60)
at org.nd4j.jita.workspace.CudaWorkspace.alloc(CudaWorkspace.java:198)
at org.nd4j.jita.allocator.impl.AtomicAllocator.allocateMemory(AtomicAllocator.java:454)
at org.nd4j.jita.allocator.impl.AtomicAllocator.allocateMemory(AtomicAllocator.java:399)

o.n.l.a.o.e.DefaultOpExecutioner - Backend used: [CUDA]; OS: [Linux]
o.n.l.a.o.e.DefaultOpExecutioner - Cores: [8]; Memory: [7.2GB];
o.n.l.a.o.e.DefaultOpExecutioner - Blas vendor: [CUBLAS]
o.n.l.j.o.e.CudaExecutioner - Device Name: [GeForce GTX 1060 3GB]; CC: [6.1]; Total/free memory: [3164733440]

androuino

018-08-07 16:24:54.449 [I] TrainObjectDetection (train:154) Train model...
2018-08-07 16:24:54.950 [E] Main (onTrain:60) Unable to train
2018-08-07 16:24:54.992 [V] Main (onTrain:60) STACK START ------------------------------------------------------------------------------
2018-08-07 16:24:54.993 [V] Main (onTrain:60)   Invalid indices: cannot get [3,0,15,7] from a [8, 6, 13, 13] NDArray
2018-08-07 16:24:54.994 [V] Main (onTrain:60)   java.lang.IllegalArgumentException: Invalid indices: cannot get [3,0,15,7] from a [8, 6, 13, 13] NDArray
2018-08-07 16:24:54.995 [V] Main (onTrain:60)   at org.nd4j.linalg.api.shape.Shape.getOffsetUnsafe(Shape.java:946)
2018-08-07 16:24:54.996 [V] Main (onTrain:60)   at org.nd4j.linalg.api.ndarray.BaseNDArray.putScalar(BaseNDArray.java:1426)
2018-08-07 16:24:54.997 [V] Main (onTrain:60)   at org.datavec.image.recordreader.objdetect.ObjectDetectionRecordReader.label(ObjectDetectionRecordReader.java:232)
2018-08-07 16:24:54.998 [V] Main (onTrain:60)   at org.datavec.image.recordre

androuino

train.yolo.classes=2
train.yolo.channels=3
train.yolo.width=416
train.yolo.height=416
train.yolo.boxes=5
train.yolo.grid.width=13
train.yolo.grid.height=13
train.yolo.batch=8
train.yolo.epochs=10000

androuino

package org.deeplearning4j.examples.convolution.objectdetection;

import java.io.File;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;
import java.util.Random;
import org.bytedeco.javacv.CanvasFrame;
import org.bytedeco.javacv.OpenCVFrameConverter;
import org.datavec.api.io.filters.RandomPathFilter;

androuino

package org.deeplearning4j.examples.convolution.objectdetection;

import java.io.File;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;
import java.util.Random;
import org.bytedeco.javacv.CanvasFrame;
import org.bytedeco.javacv.OpenCVFrameConverter;
import org.datavec.api.io.filters.RandomPathFilter;

androuino

package org.deeplearning4j.examples.convolution.objectdetection;

import java.io.File;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import org.bytedeco.javacv.CanvasFrame;
import org.bytedeco.javacv.OpenCVFrameConverter;

androuino

{
  "nodes": [
    {
      "op": "null", 
      "name": "data", 
      "inputs": []
    }, 
    {
      "op": "null", 
      "name": "darknetv30_conv0_weight", 

androuino

Exception in thread "main" java.lang.UnsatisfiedLinkError: /tmp/mxnet463915887770065837/mxnet-scala: libcudart.so.9.2: cannot open shared object file: No such file or directory
	at java.lang.ClassLoader$NativeLibrary.load(Native Method)
	at java.lang.ClassLoader.loadLibrary0(ClassLoader.java:1941)
	at java.lang.ClassLoader.loadLibrary(ClassLoader.java:1824)
	at java.lang.Runtime.load0(Runtime.java:809)
	at java.lang.System.load(System.java:1086)
	at org.apache.mxnet.util.NativeLibraryLoader$.loadLibraryFromFile(NativeLibraryLoader.scala:131)
	at org.apache.mxnet.util.NativeLibraryLoader$.loadLibrary(NativeLibraryLoader.scala:99)
	at org.apache.mxnet.Base$.<init>(Base.scala:72)
	at org.apache.mxnet.Base$.<clinit>(Base.scala)

androuino

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *

androuino

"""Train YOLOv3 with random shapes."""
import argparse
import os
import logging
import time
import warnings
import numpy as np
import mxnet as mx
from mxnet import nd
from mxnet import gluon