コード例 #1
0
        public override int Run(string[] remainingArguments)
        {
            var trainable = YOLO.CreateV4Trainable(inputSize: this.InputSize,
                                                   classCount: this.ClassCount,
                                                   strides: this.Strides);

            trainable.load_weights(this.WeigthsPath);
            var    output  = YOLOv4.Output.Get(trainable);
            Tensor input   = trainable.input_dyn;
            var    savable = YOLO.CreateSaveable(inputSize: this.InputSize, input, output,
                                                 classCount: this.ClassCount,
                                                 strides: this.Strides,
                                                 anchors: tf.constant(this.Anchors),
                                                 xyScale: YOLOv4.XYScale,
                                                 scoreThreshold: this.ScoreThreshold);

            savable.summary();
            savable.save(this.OutputPath, save_format: "tf", include_optimizer: false);
            return(0);
        }