如何使用darknet架构和fastai进行图像分类

2024-06-26 08:42:18 发布

您现在位置:Python中文网/ 问答频道 /正文

例如,我可以使用squezenet进行训练,但我是新手,所以我无法确定其他架构(如darknet、exception或xresnet)需要成功实例化什么。在

在我的笔记本里我试着:

arch = models.Darknet(num_classes = 2, num_blocks = 4) // whats numblocks
learn2 = cnn_learner(data = data, base_arch = arch , metrics=[accuracy, error_rate], pretrained = True)

我得到错误:

^{pr2}$

Tags: 实例datamodels架构exception笔记本numclasses
1条回答
网友
1楼 · 发布于 2024-06-26 08:42:18

docs

Create a Darknet with blocks of sizes given in num_blocks, ending with num_classes and using nf initial features. Darknet53 uses num_blocks = [1,2,8,8,4].

了解有关darknet的更多信息here

相关问题 更多 >