tacode / pspnet_voc Goto Github PK
View Code? Open in Web Editor NEWPytorch implementation of PSPNet on VOC
Pytorch implementation of PSPNet on VOC
Sorry to disturb,can I have your pre-trained model?The model I download from hsZhao can not match this
Do you forget to update the learning rate when you reuse the snapshot? I mean, although you reuse the loss and epoch, but the learning rate is still start-lr.
Hi, I tried running the train function and came across this error
File "train.py", line 150, in <module>
train()
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/click/core.py", line 829, in __call__
return self.main(*args, **kwargs)
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/click/core.py", line 782, in main
rv = self.invoke(ctx)
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/click/core.py", line 1066, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/click/core.py", line 610, in invoke
return callback(*args, **kwargs)
File "train.py", line 114, in train
for i,(x, y, y_cls) in enumerate(train_loader):
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 345, in __next__
data = self._next_data()
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 385, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/scratch/gd1302/miniconda3/envs/cv/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/scratch/gd1302/pspnet/PSPNet_VOC/voc_loader.py", line 101, in __getitem__
cls = self.getcls(lbl)
File "/scratch/gd1302/pspnet/PSPNet_VOC/voc_loader.py", line 108, in getcls
cls = np.zeros((21),dtype=int)
IndexError: index 21 is out of bounds for axis 0 with size 21
Do you know what could be causing this?
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.