飞飞君

Pytorch 官方代码 文章主要为了去除官方代码中多余代码，保留能实现程序的最小代码 使用pytorch框架，maskrcnn或fastrcnn(不完全)实现 头文件导入 import os import numpy as np import torch from PIL import Image from engine import train_one_epoch, evaluate import utils import torchvision from torchvision.models.dete

文章主要包含：官方数据集导入、自定义数据集，自定义网络结构，训练，训练后的模型使用 头文件导入 import torch import torchvision import torchsummary import os import numpy as np import matplotlib.pyplot as plt 常量定义 BATCH_SIZE = 64 #图像行列像素数量 IMAGE_ROW = 28 IMAGE_COL = 28 #数据根路径 DATA_SOURCE_DIR