SPN.pytorch
SPN.pytorch copied to clipboard
yeezhu
PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.
(py3.5) ltc@ltcpc:~/PycharmProjects/SPN.pytorch-master/spnlib$ python setup.py Including CUDA code. usage: setup.py [global_opts] cmd1 [cmd1_opts] [cmd2 [cmd2_opts] ...] or: setup.py --help [cmd1 cmd2 ...] or: setup.py --help-commands or: setup.py cmd --help error: no...
``` model_path = os.path.join(DATA_ROOT, 'SP_GoogleNet_ImageNet.pt') model_dict = utils.load_model_imagenet(model_path) ``` ``` --------------------------------------------------------------------------- ValueError Traceback (most recent call last) in () 1 model_path = os.path.join(DATA_ROOT, 'SP_GoogleNet_ImageNet.pt') ----> 2 model_dict = utils.load_model_imagenet(model_path) ~/choas/method/SPN.pytorch/demo/experiment/util.py...
Hi @yeezhu, Thanks very much for the nice research and implementation. Looks like pytorch has deprecated support for c extensions: https://gitlab.com/elynn/pytorch/commit/15d28e400f71c50848b760ca188ebc885ee433d8 . Would like to request conversion to c++ so...
Hi, @yeezhu. I have some questions on experiments in the paper: 1. In section 4.2, Two different methods ("pointing without prediction" and "pointing with prediction") are used to to evaluate...
Hi, Nice Work. But I think there is actually no need for CUDA code -- they could be implemented vectorized, which is more intuitive. For example: [`__global__ void InitDistanceMetricKernel`](https://github.com/yeezhu/SPN.pytorch/blob/f8ebdca128ce6b550d090b45dae7ef43d54d00e1/spnlib/spn/src/libspn_kernel.cu#L4) could...
I got these warnings during installation. I went through [a closed issue](https://github.com/yeezhu/SPN.pytorch/issues/2) that showed the same warnings. I upgraded pytorch to pytorch 0.3.0, but the warning still exist. It looks...
