pytorch
lzAllen
这个作者很懒,什么都没留下…
展开
-
跑通MMDetection的MaskRCNN
初次跑通MMDetection的MaskRCNNDemo惯例先跑一下demo,新建一个HellowMMDection.py,将这个脚本放在./demo中去:# Check Pytorch installationimport torch, torchvisionprint(torch.__version__, torch.cuda.is_available())# Check MMDetection installationimport mmdetprint(mmdet.__ver原创 2020-11-02 20:33:54 · 2348 阅读 · 3 评论 -
标签平滑的pytorch的简单实现
标签平滑的pytorch的简单实现import torchimport torch.nn as nndef CrossEntropyLoss_label_smooth(outputs, targets, num_classes=10, epsilon=0.1): N = targets.size(0) # torch.Size([8, 10]) # 初始化一个矩阵, 里面的值都是epsilon / (num_classes - 1) smoothed_labels原创 2020-11-01 16:21:20 · 2893 阅读 · 0 评论