摘要: def build_ssd(phase, size=300, num_classes=21): if phase != "test" and phase != "train": print("ERROR: Phase: " + phase + " not recognized") return if 阅读全文
posted @ 2021-06-10 15:37 无左无右 阅读(575) 评论(0) 推荐(0)
摘要: https://github.com/sksq96/pytorch-summary 安装 pip install torchsummary 使用 from torchsummary import summary summary(your_model, input_size=(channels, H, 阅读全文
posted @ 2021-06-10 10:03 无左无右 阅读(729) 评论(0) 推荐(0)