search

dotnet / TorchSharp

A .NET library that provides access to the library that powers PyTorch.
MIT License
1.32k stars 169 forks source link

Is it possible to implement Faster rcnn? #750

Open xhuan8 opened 1 year ago

xhuan8 commented 1 year ago

I try to use Faster rcnn for object detection, training and inference. Is it possible to implement faster rcnn? How much work is required?

NiklasGustafsson commented 1 year ago

It would take me a few days to implement based on the torchvision code, which isn't too bad.

Testing would be the real challenge -- I don't think that I have the data to test it properly. This is the kind of thing that would be perfect for someone who is knowledgeable about the domain to help out with.

NiklasGustafsson commented 1 year ago

@xhuan8 -- I see that you have been working on things in your private fork. Is there anything ready to PR to dotnet/TorchSharp?

xhuan8 commented 1 year ago

I haven't finished the testing, the torchvision dll seems not working, maybe need implement custom roi_align method. also the packing has some problem on my local machine.

I'm a kind of busy recently, will continue the work on the middle of next month.