pytorch-segmentation-detection
pytorch-segmentation-detection copied to clipboard
Image Segmentation and Object Detection in Pytorch
Hi, I found your flops benchmark quite useful. Is it possible to add a license file to your project, (may be similar to the one you have used in another...
Hi, I try to train model using python 3, but I got below issue: ``` /home/v2m/anaconda3/envs/my_env3/lib/python3.7/site-packages/torch/nn/_reduction.py:46: UserWarning: size│ _average and reduce args will be deprecated, please use reduction='sum' instead. │...
Hello, Can I know the optimizer and its specifications to use on unet model on Pascal Voc segmentation using FOCAL loss ? Should I have to use any learning rate...
@warmspringwinds I am getting the following error for resnet_18_8s_59.pth RuntimeError: Error(s) in loading state_dict for Resnet18_8s: size mismatch for resnet18_8s.fc.bias: copying a param with shape torch.Size([2]) from checkpoint, the shape...
By using your fork of `torchvision` and default installation of pytorch for `Linux-Python3.6-CUDA10`: 1. `init_weights` argument in the class `VGG` was missing. 2. After fixing (1), the following error was...
``` TypeError: torch.FloatTensor constructor received an invalid combination of arguments - got (int, int, numpy.int64, numpy.int64), but expected one of: * no arguments * (int ...) didn't match because some...
While executing any of the notebooks in: `pytorch_segmentation_detection/recipes/pascal_voc/segmentation/*.ipynb` the errors relate to initialization regarding unknown arguments for ResNet object init. Tried on multiple cloud / GPU setups with same output....
I'm new to implementing CNNs and I'm trying to understand how a model knows whether to perform semantic classification (pixelwise) or image classification (one class per image). As far as...