螺丝松紧分析-5

Posted on Mar 15, 2025

Table of contents:

数据检查

通过下面的脚本，检查一下数据量。

find . -type d ! -path "*/.*" | while read folder
do
    # 忽略隐藏文件夹中的文件，统计当前目录内的 .png 文件数量
    count=$(find "$folder" -maxdepth 1 -type f -iname "*.png" ! -iname ".*" | wc -l)
    # 根据层级生成树状结构
    indent=$(echo "$folder" | awk -F'/' '{for(i=2;i<NF;i++) printf "│   "; if(NF>1) printf "├── "}')
    # 输出目录及文件数量
    echo "${indent}${folder##*/} ($count)"
done

可以检查一下下面 patient 和 model 用于训练和验证的数据量是否准确，以及进入文件夹里做一下基础的数据验证。

# patient
├── loose (121)
├── tight (720)
├── train (0)
│   ├── loose (95)
│   ├── tight (575)
├── val (0)
│   ├── loose (13)
│   ├── tight (72)
├── test (0)
│   ├── loose (13)
│   ├── tight (73)

# model
├── loose (465)
├── tight (547)
├── train (0)
│   ├── loose (369)
│   ├── tight (433)
├── val (0)
│   ├── loose (49)
│   ├── tight (59)
├── test (0)
│   ├── loose (47)
│   ├── tight (55)

结果分析

从训练的过程看，大部分都是正常的。

从结果看，inceptionv3-fine-tuned 的结果最好。

验证集的 confusion matrix 也是比较好的。