PyTorch
Wayne2019
这个作者很懒,什么都没留下…
展开
-
AI challenger 场景分类 PyTorch 迁移学习 resnet18
没计算资源,只能简单测试下resnet18训练代码(带验证):'''TODO:- 采用更深的网络(简单,但是需要计算资源)- top3 accuracy,可参考https://github.com/pytorch/examples/blob/master/imagenet/main.py'''#pkill -9 python#nvidia-smiimport osimport tor原创 2017-09-26 20:13:41 · 3987 阅读 · 0 评论 -
AI challenger 场景分类 PyTorch 迁移学习 Places365-CNNs 启动代码
分享个简单的启动代码。'''CHANGES:- imagenet cnns: resnet: http://pytorch.org/docs/master/torchvision/models.html- places 365 cnns: resnet 18, 50: https://github.com/CSAILVision/places365- top3 accuracy: https原创 2017-10-13 10:28:04 · 5720 阅读 · 3 评论 -
AI challenger 场景分类 PyTorch 测试代码
- 有个比较奇怪的地方是loss和batch_size有关,目前没有仔细研究。 - 另外需要注意places 365 cnn 是采取了保存和加载整个模型的办法,而我自己在用时只保存了权重,因此这里需要先导入pytorch官方的模型。 - 经在验证集上多重的比较,除了loss的问题其他都正常。 - 只是为了测试程序功能,使用了最小的训练量,因此准确率非常低。'''http://blog.csd原创 2017-10-14 08:15:56 · 2340 阅读 · 4 评论 -
Ai challenger 场景分类: 观察验证集中的错误分类情况
基于官方的验证脚本改的。import jsonimport argparseimport timeimport shutilimport pandas as pddef __load_data(submit_file, reference_file, submit_dict, ref_dict): # load submit result and reference result原创 2017-11-03 10:52:12 · 637 阅读 · 0 评论