Sovit Ranjan Rath
Sovit Ranjan Rath
It may be dataset specific. Kind of hard to say much here actually. In fact, this is the first time I am seeing this error. Still, will try to research...
@Arunxavierpro You can pass the desired weight path using `--weights` flag. e.g. `python train.py --weights weights.pth ` You can also pass `--resume-training` (boolean flag) if you want to continue with...
Please close the issue from your side if this solves your query.
It depends on your commands also. But in general, it should be: `python train.py --config data_configs/smoke.yaml --epochs 100 --model fasterrcnn_resnet50_fpn --project-name smoke_training --batch-size 16`
@Arunxavierpro With the above command you need to pass the weights file. `python train.py --config data_configs/smoke.yaml --epochs 100 --model fasterrcnn_resnet50_fpn --project-name smoke_training --batch-size 16 --weights weights.pth --resume-training`
@Arunxavierpro My bad. I forgot to mention. When you pass `--resume-training`, you need to provide more number of epochs than the previous training. It will resume from the previous training....
@kalikhademi Right now, the code computes only average precision and recall. While executing `eval.py`, if you pass `--verbose`, it will show you class-wise precision also. Regarding the TP, FP, FN,...
@samahwaleed After `eval.py` finishes executing, the recall will be automatically printed.
@samahwaleed Yes, right now, it's AP for each class and the overall recall. I will try to add recall for each class as well.
@samahwaleed Sure, I will try my best and update here.