search

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
642 stars 259 forks source link

Can I use this code in a Google Colaboratory? #62

Closed RickStor closed 4 years ago

RickStor commented 4 years ago

Awesome work!

I´m trying to use the code in a Google Colaboratory. But I have this issue:

ImportError Traceback (most recent call last)

in () 11 from matplotlib import cm as CM 12 from image import * ---> 13 from model import CSRNet 14 import torch 15 get_ipython().magic('matplotlib inline') /content/model.py in () 2 import torch 3 from torchvision import models ----> 4 from utils import save_net,load_net 5 6 class CSRNet(nn.Module): ImportError: cannot import name 'save_net'

Could you help me please?