小kk_p - 博客园

2022年3月

摘要：第一步：模型转换，按照github一步一步来就ok了~此处无坑第二步：cmake建立vs工程，需要在cmakelist里面需要使用的accelerator，否则在getdevice会返回NULL值第三步：调用 #include "tnn/utils/dims_vector_utils.h" #i 阅读全文

posted @ 2022-03-18 22:14 小kk_p 阅读(477) 评论(0) 推荐(0)

cmake和CMakeLists.txt的学习

摘要： https://www.jianshu.com/p/cb4f8136a265 阅读全文

posted @ 2022-03-18 11:11 小kk_p 阅读(27) 评论(0) 推荐(0)

classify

摘要： https://blog.csdn.net/didiaopao/article/details/120717525 https://zhuanlan.zhihu.com/p/81688220 阅读全文

posted @ 2022-03-15 14:43 小kk_p 阅读(49) 评论(0) 推荐(0)

获取tensor的非重复数字

摘要： np.unique(labels.cpu().detach().numpy()) 阅读全文

posted @ 2022-03-15 09:34 小kk_p 阅读(116) 评论(0) 推荐(0)

Invalid shape (3, 1024, 2048) for image data

摘要：仅作为记录，大佬请跳过。用.T 即 (3, 1024, 2048)不能plt，而应该是(某,某,3)才能显示。阅读全文

posted @ 2022-03-15 09:24 小kk_p 阅读(889) 评论(0) 推荐(0)

torch.tensor转numpy

摘要： outputs.cpu().detach().numpy()[0,:,:,:].shape import matplotlib.pyplot as pyplot pyplot.imshow(binary_mask)pyplot.show() 阅读全文

posted @ 2022-03-15 09:21 小kk_p 阅读(395) 评论(0) 推荐(0)

豆瓣源

摘要： pip install xxx -i https://pypi.douban.com/simple/ pip install -r req.txt -i https://pypi.douban.com/simple/ 阅读全文

posted @ 2022-03-14 20:29 小kk_p 阅读(102) 评论(0) 推荐(0)

tensor转为 numpy.array的格式

摘要： outputs[0,0,:,:].cpu().detach().numpy() 阅读全文

posted @ 2022-03-14 11:52 小kk_p 阅读(217) 评论(0) 推荐(0)

轻量级语义分割网络：ENet

摘要： https://github.com/davidtvs/PyTorch-ENet 阅读全文

posted @ 2022-03-13 17:31 小kk_p 阅读(129) 评论(0) 推荐(0)

将数据集json格式转为coco格式

摘要： https://blog.csdn.net/qq_34097521/article/details/101191800 json2coco img_dir 为原始图片文件夹anno_dir为数据集的json格式格式如下 [ { "name": "d6718a7129af0ecf0827157752. 阅读全文

posted @ 2022-03-13 15:25 小kk_p 阅读(651) 评论(0) 推荐(0)

公告