[人工智能]从无到有跑通KAPAO

kapao工程：https://github.com/wmcnally/kapao

论文地址：https://arxiv.org/abs/2111.08557

由于之前没有接触过关键点的工作，所以我选择先下载作者提到的数据，将数据按照他要求的方式存放，然后通过运行的代码来查看整个工程的结果；

使用kapao进行托盘目标检测和托盘目标的四个角点的检测；原理都还没有来得及查看，然后直接进行模型训练；

处理过程的整体流程：

1.将整个样本分成训练集和验证集合，通过txt文件，记载相关的的名字；

"""
将所有样本分拣为训练样本和测试样本
"""
import argparse
import os
import math
import random
import shutil
from tqdm import tqdm

class Process:
    def __init__(self,args):
        self.imgs_dir = args.imgs_dir
        self.save_dir = args.save_dir
        self.ratio = args.ratio

    def check_path(self, chpath):
        if not os.path.exists(chpath):
            os.mkdir(chpath)

    def check_path

开通VIP 解锁文章