Comments (3)
Hi @sushilkhadkaanon, thanks for your interest in our work.
I believe the instructions are comprehensive enough, as other users (#17, #31) have been able to train on custom datasets following my instructions. I suggest you spend some more time trying to understand registering custom datasets in Detectron2 and then try preparing a custom dataset mapper.
Let me know if you have any specific issues.
from oneformer.
Hi! Thanks, I really appreciate your work. I tried following your instructions.
I added InstanceCOCOCustomNewBaselineDatasetMapper in build_train_loader of train_net.py
But where do I register my dataset?
` #
from detectron2.data.datasets import register_coco_instances
register_coco_instances("my_dataset_train", {}, "_annotations.coco.json", "custom_data/train")
register_coco_instances("my_dataset_test", {}, "_annotations.coco.json", "custom_data/train")
metadata_train = MetadataCatalog.get("my_dataset_train")
cfg.DATASETS.TRAIN = ("my_dataset_train",)
cfg.DATASETS.TEST = ("my_dataset_test",)
cfg.SOLVER.MAX_ITER = 5
`
Added this inside main function in train_net.py.
Now this is giving me `ModuleNotFoundError:
Please compile MultiScaleDeformableAttention CUDA op with the following commands:
\`cd mask2former/modeling/pixel_decoder/ops\`
\`sh make.sh\``
from oneformer.
You can look at register_ade20k_instance.py
for reference. You need to create a similar file for your dataset. Also, the official detectron2 guide on adding custom datasets is quite detailed.
from oneformer.
Related Issues (20)
- undefined symbol: _ZNK3c107SymBool10guard_boolEPKcl
- whether I need to download Swin transformer pretraining weight for training
- Swin large backbone warning: "norm.bias will not be loaded. Please double check and see if this is desired." HOT 1
- AssertionError when trying to reproduce result HOT 2
- Inference time
- Class labels Issue
- Different result in different GPU
- Update colab demo HOT 1
- Resources required to fine-tune this model with swin-l HOT 13
- Potential code bug but always good model
- Freezing layers for DiNAT model
- Cityscapes val SOTA results not found in the paper HOT 2
- Training ADE20K for Instance Segmentation Only
- ModuleNotFoundError: No module named 'detectron2.config' HOT 3
- Wrong citation
- What is the expected dimension sizes for the outputs dictionary from sem_seg_head? HOT 3
- Installation and setting up this repo is challenging HOT 1
- How to set prefetch_factor? HOT 1
- How to set the image size to run demo ?
- Runtime Error HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from oneformer.