View Code? Open in Web Editor
NEW
本仓库将使用Pytorch框架实现经典的图像分类网络、目标检测网络、图像分割网络,图像生成网络等,并会持续更新!!!
Python 25.60%
Jupyter Notebook 74.40%
cv's Introduction
- 🔭 I’m currently in Beijing, China
- 📫 How to reach me:
[email protected]
- 🌱 I’m currently working on low-level Computer Vision and Medical Image Segmentation.
cv's People
Contributors
Watchers
cv's Issues
您好,在pycharm依照配置文件安装环境时出现torch、torchvision和opencv无法安装成功的情况。用anaconda控制台安装显示pytorch和torchvision与其他环境有冲突?
博主 请问如果要选择其中之一作为backbone进行训练得到预训练权重文件,是不是只要把输出通道数和分类的代码改掉就可以了?
SegNet的网络如果根据网络结构图来看,Encoder345和Decoder123是三层卷积+BN+ReLU吧。
您好,我在model_path导入ep050.pth的文件,但是他出现错误:KeyError: 'module.encoder.encode1.0.weight'
同时,
ep050.pth在训练集与测试集loss均达0.05,但是,我用它predict的时候,只有一种颜色输出,甚至用训练集里的照片也是这样的结果
想请教一下:
请问是模型保存的出现问题了吗?
或者是torch.nn.DataParallel的问题吗?(我的cuda定义是True,但是我只有一张显卡;可是我cuda定义是False时是cpu训练,很慢)
[train epoch 0] loss: 0.339386, acc: 7.600000: 100%|
[valid epoch 0] loss: 0.001, acc 1.000: 100%|
[train epoch 1] loss: 0.001157, acc: 8.000000: 100%
这准确率是多少啊?100%?还是说没显示?