Validate Classification Model - Darknet Validation Command Line

版权声明:世上没有白读的书,每一页都算数。 https://blog.csdn.net/chengyq116/article/details/86103449

Validate Classification Model - Darknet Validation Command Line

1. darknet_validation_command_line_classification.py

#
# Created by foreverstrong on 5/30/2018
#

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"

## network_resolution = 288
# network_resolution = "288"

## network_resolution = 320
# network_resolution = "320"

## network_resolution = 352
# network_resolution = "352"

## network_resolution = 384
# network_resolution = "384"

## network_resolution = 416
# network_resolution = "416"

gpu_index = 0
min_iteration = 0

weights_path = "/media/famu/DISK_DATA/yongqiang/backup_tiny_darknet_autoparts_001"
command_line_base = "./darknet classifier valid ./cfg/tiny_darknet_autopart_001.data ./cfg/tiny_darknet_autopart_001.cfg -input_dir=.weights -i %s" % (
    gpu_index)


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

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

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

    num = 0

    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')
            num = num + 1

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


if __name__ == "__main__":
    darknet_validation_command_line(weights_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

猜你喜欢

转载自blog.csdn.net/chengyq116/article/details/86103449