CenterNet 读书笔记

 

这个是训练人体的：

https://github.com/ZongweiZhou1/CenterNetPerson

 

人体+跟踪

https://github.com/kimyoon-young/centerNet-deep-sort

人脸的：只有测试和模型

https://github.com/smartwell/centernet-train-wider-face

 

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

import torch.utils.data as data
import numpy as np
import torch
import json
import cv2
import os
from utils.image import flip, color_aug
from utils.image import get_affine_transform, affine_transform
from utils.image import gaussian_radius, draw_umich_gaussian, draw_msra_gaussian
from utils.image import draw_dense_reg
import math

class CTDetDataset(data.Datase