-
### 🐛 Describe the bug
the model only runs if the batch size is > 1 e.g. image_input = torch.randn(2, 3, 224, 224), is this intended?
Using (1, 3, 224, 224) results in a ShapeProp error. Here's t…
-
Here is my code.
`import torch
import torch.onnx
from retinanet import model
model= model.resnet50(num_classes=80,)
model.load_state_dict(torch.load('../models/coco_resnet_50_map_0_335_state_di…
-
YALTAi 2.0.1 hangs forever when segmenting
```
(train-2.0.1-py3.11) incognito@DESKTOP-H1BS9PO:~/YALTAi$ yaltai kraken -I "*.jpg" --suffix ".xml" segment --yolo runs/detect/train2/weights/best.pt
…
-
## 🐛 Bug
I have followed the README in examples/speech_to_text to reproduce results of ST on MUSTC. But when I start training (after preprocess following the instruction), the system raises segment…
-
Hi! Thank you for your awesome code. I have a question:
Suppose I have a set of volume data. How can I use the pretrained model to segment the ER and mito. I'm a little bit confused about the evaluat…
-
These are the results of the current segnet implementation:
![horse](https://cloud.githubusercontent.com/assets/1780466/25223935/2d0c605e-25bd-11e7-8a0a-cd23f793f32e.png)
![horse-segnetfix](https://…
-
I am training a version of unet with joint classification and semantic segmentation using `O1` level. The training crashes after I explicitly cast `box_coord_tensor` in `roi_pool` function.
```python…
-
### Bug description
I'm trying to port TorchGeo to LightningCLI: https://github.com/microsoft/torchgeo/pull/1559
99% of things are great! Unfortunately I'm still stuck on the last 1%. If you loo…
-
We currently have very limited testing for the models in `torchvision`.
In most cases, we only test for the output shape to be correct, see
https://github.com/pytorch/vision/blob/8635be94d1216f10f…
-
My programs crash with a segmentation fault when trying to initialize a model with `create_model_and_transforms` on an Intel Mac. I have tried different devices (cpu, mps) and different models - the c…