Function main

lesson6-Segmentation/main.cpp:4–16

#include"Segmentor.h"

int main(int argc, char *argv[])
{
    cv::Mat image = cv::imread("D:\\AllentFiles\\data\\dataset4teach\\voc_person_seg\\val\\2007_004000.jpg");

    Segmentor<UNet> segmentor;
    segmentor.Initialize(0,512,512,{"background","person"},
                         "resnext50_32x4d","D:\\AllentFiles\\code\\tmp\\resnext50_32x4d.pt");
    segmentor.LoadWeight("segmentor.pt");
    segmentor.Predict(image,"person");
    segmentor.Train(0.003,300,4,"D:\\AllentFiles\\data\\dataset4teach\\voc_person_seg",".jpg","segmentor.pt");

    return 0;
}