图像分类问题中正负样本如何分类 40C

假设我们要处理图像分类问题,假设batch_size=16,num_class=10,那么进行on

e-hot编码之后的label为[16,10]的张量,而经过model后的output也应该为[16,10],假如我现在要获得正样本,负样本,应该如何用代码去实现。

目的是为了使用torch.nn.TripletMarginLoss,其要求输入正样本和负样本:
图片说明

1个回答

你这个只是完成了模型的搭建,下面需要训练,然后才能预测
在backward下面依次需要定义
optimizer,比如说sgd adam等
然后定义step,最后train才是学习。

https://blog.csdn.net/andyL_05/article/details/103363603

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问