![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch API
记录一下相应的API,以及对应的例程
一名CV界的小学生
这个作者很懒,什么都没留下…
展开
-
打印网络名称以及相应shape
【代码】打印网络名称以及相应shape。原创 2023-07-21 15:04:09 · 68 阅读 · 0 评论 -
yoloV4模型转onnx
def pth2onnx(cfgDir,pthDir,imgSize): model=Darknet(cfgDir,imgSize) checkpoint = torch.load(pthDir) print(model) model.load_state_dict(checkpoint['model']) model.eval() dummy_input = torch.randn(1, 3, 416, 416) input_names = ["i.原创 2022-01-25 19:58:14 · 2554 阅读 · 0 评论 -
pytorch API:随机裁剪
def imgAugTest(imgDir): img=Image.open(imgDir) print('the size of raw img:{}'.format(img.size)) cropImg=transforms.RandomResizedCrop(224)(img) print('the size of new img {}'.format(cropImg.size)) cropImg01=transforms.RandomResizedCrop(.原创 2021-12-17 15:46:04 · 1025 阅读 · 0 评论 -
Pytorch API:学习率调整
def StepLRTest(): model=AlexNet(num_classes=2) optimizer=optim.SGD(params=model.parameters(),lr=0.01) scheduler=lr_scheduler.MultiStepLR(optimizer,milestones=[5,20,40,80],gamma=0.1) plt.figure() x=list(range(100)) y=[] for ep.原创 2021-12-13 16:32:38 · 1394 阅读 · 0 评论 -
pytorch: 查看特征图
其中网络结构为:GitHub - chenhuanan/person_search_demo: 利用YOLOv3结合行人重识别模型,实现行人的检测识别,查找特定行人from reid.modeling import build_modelfrom reid.config import cfg as reidCfgimport torchimport cv2import numpy as npimport torchvision.transforms as transformsimpo.原创 2021-09-23 15:10:17 · 503 阅读 · 1 评论