Comments (8)
Hi - I will have a quick look, but note my paper does not cover anomaly detection across RGB images which can be analysed further in future work.
from anoddpm.
Did the model train ok?
How were the output images / videos?
The error you have sent is caused in the model loading process not in the detection process - which I would struggle to test myself as I don't have a dedicated GPU (and I no longer have access to the university server which trained my models)
from anoddpm.
I can't seem to replicate the problem on my end - I've been able to load in a checkpoint with those arguments - it would be great if you could provide more information
from anoddpm.
I am running the run with RGB set to True, do you think that could be affecting it?
Thanks for the solution!
Perhaps TRAIN is working well. The reason for this is that, as you can see in the image below, a movie is created that generates an image of the original leather dataset from the noise.
I am running the run with RGB set to True, do you think that could be affecting it?
from anoddpm.
I am running the run with RGB set to True, do you think that could be affecting it? Thanks for the solution! Perhaps TRAIN is working well. The reason for this is that, as you can see in the image below, a movie is created that generates an image of the original leather dataset from the noise.
I am running the run with RGB set to True, do you think that could be affecting it?
How long to train an Epoch? I'm stuck at the first epoch for many hours
And if possible, could you share your python environment?
from anoddpm.
Hi I tried to run the introduced model using MVtec dataset according to READ.ME, but I got the following error in detection.py. I don't know how to solve this problem. Please help me to solve this problem.
The test_arg I used is as follows
{ "img_size": [ 256, 256 ], "Batch_Size": 1, "EPOCHS": 3000, "T": 1000, "base_channels": 128, "beta_schedule": "linear", "channel_mults": "", "loss-type": "l2", "loss_weight": "none", "train_start": true, "lr": 1e-4, "random_slice": true, "sample_distance": 600, "weight_decay": 0.0, "save_imgs": false, "save_vids": true, "dropout": 0, "attention_resolutions": "16,8", "num_heads": 2, "num_head_channels": -1, "noise_fn": "gauss", "dataset": "leather", "channels": 3 }
The error statement is as follows.
RuntimeError: Error(s) in loading state_dict for UNetModel: size mismatch for down.0.0.weight: copying a param with shape torch.Size([128, 3, 3, 3]) from checkpoint, the shape in current model is torch. size mismatch for out.2.weight: copying a param with shape torch.Size([3, 128, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([1, 128, 3, 3]). size mismatch for out.2.bias: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([1]).
Waiting for a solution.
Do you know how to use detection.py? I want to get the result image of detection through this file, but I don't know how to get this image
I would really appreciate it if I could get a reply
from anoddpm.
Hi,
Apologies for not replying sooner, I've been incredibly busy and I'm the sole contributor to this repo.
I am running the run with RGB set to True, do you think that could be affecting it? Thanks for the solution! Perhaps TRAIN is working well. The reason for this is that, as you can see in the image below, a movie is created that generates an image of the original leather dataset from the noise.
I am running the run with RGB set to True, do you think that could be affecting it?
It will be affecting it, I wrote bits of code to test this when I was at the late end of writing the paper and likely have not properly reverted it. Running the code with RGB channels will require a bit more work to revert this back to fully working; I don't quite have enough time now to sort this (plus I don't have the same hardware anymore).
How long to train an Epoch? I'm stuck at the first epoch for many hours And if possible, could you share your python environment?
A GPU is basically required for a model of this size, a single iteration (not epoch) on my MacBook Pro (running with cpu) does not even finish in an hour. From memory, the paper was generated on an NVIDIA RTX 2080 series (I believe it's noted in the paper). I believe I provided a requirements.txt to another issue if you wish to view that.
Do you know how to use detection.py? I want to get the result image of detection through this file, but I don't know how to get this image I would really appreciate it if I could get a reply
If all you would like is result images; generate_images.py is far easier to understand and does exactly that. detection.py was primarily used for metric gathering and consequently looping across the whole dataset and concatenating results to form AUC, dice and iou graphs.
Does this help?
from anoddpm.
Closing due to no responses
from anoddpm.
Related Issues (20)
- some doubts about the principle of the diffusion model HOT 1
- some quesion about detection HOT 5
- some questions about DDPM HOT 1
- Some problems with running detection HOT 2
- Batch Size HOT 3
- Where is the file of args used in the paper? HOT 2
- How Train and Test is split. HOT 2
- comparasion_models HOT 5
- What is a directory raw_cleaned ? HOT 7
- about the training time cost HOT 1
- Some problems about data HOT 1
- Encountered Access Issue of E-mail HOT 1
- Package Versions / Install Requirements? HOT 1
- how to get testData HOT 2
- high CPU usage discovered in training stage HOT 1
- result for MVTec HOT 1
- Asking about "import Comparative_models.CE as CE" HOT 2
- Problem about saving .mp4 files during training process HOT 3
- Access to training args.json used in the paper
- Pre-trained data 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 anoddpm.