Validate Classification Model - Darknet Validation Command Line

Validate Classification Model - Darknet Validation Command Line

1. darknet_validation_command_line_classification.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Yongqiang Cheng

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import os

## network_resolution = 64
# network_resolution = "64"

## network_resolution = 96
# network_resolution = "96"

## network_resolution = 128
# network_resolution = "128"

## network_resolution = 160
# network_resolution = "160"

## network_resolution = 192
# network_resolution = "192"

## network_resolution = 224
network_resolution = "224"

## network_resolution = 256
# network_resolution = "256"

gpu_index = 0
min_iteration = 0

weights_file_path = "/media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001"
data_file = "tiny_darknet_autopart_001.data"
cfg_file = "tiny_darknet_autopart_001.cfg"

command_line_base = "./darknet classifier valid ./cfg/%s ./cfg/%s -input_dir=.weights -i %s" % (
    data_file, cfg_file, gpu_index)


def darknet_validation_command_line(weights_file_path, command_line_base, network_resolution, min_iteration):
    weights_dir = weights_file_path + '/'
    weights_names = os.listdir(weights_dir)
    weights_names.sort()

    runsh_path_name = ("%s/%s_valid_%s.sh" % (weights_file_path, weights_file_path.split('/')[-1], network_resolution))

    runsh_fp = open(runsh_path_name, 'w')
    runsh_fp.write("#!/bin/bash" + '\n')
    runsh_fp.write('\n')

    for weights_name in weights_names:
        if ".weights" in weights_name:
            weights_name_prefix = weights_name.split('.')[0]
            weights_num = int(weights_name_prefix.split('_')[-1])

            if weights_num < min_iteration:
                continue

            target_pathname = command_line_base.replace("-input_dir=.weights", weights_dir + weights_name)
            print(target_pathname)

            runsh_fp.write(target_pathname + '\n')
        pass
    pass

    runsh_fp.write('\n')
    runsh_fp.close()


if __name__ == "__main__":
    current_directory = os.path.dirname(os.path.abspath(__file__))
    print("current_directory:", current_directory)
    darknet_validation_command_line(weights_file_path, command_line_base, network_resolution, min_iteration)

2. Python

/usr/bin/python3.5

3. backup_tiny_darknet_autoparts_001_valid_224.sh

#!/bin/bash

./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000001.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000002.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000003.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000004.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000005.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000006.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000007.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000008.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000009.weights -i 0
./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg /media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001/tiny_darknet_autopart_001_000010.weights -i 0
......

yongqiang@famu-sys:~/darknet_work/darknet_181018/darknet$ bash backup_tiny_darknet_autoparts_001_valid_224.sh