KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.
MIT License
2.65k stars 260 forks source link

ResNet #14

Closed BZbyr closed 4 years ago

BZbyr commented 4 years ago

Using ResNet?

KichangKim commented 4 years ago

Yes. It is based on Microsoft's resnet implementation. https://github.com/microsoft/CNTK/tree/master/Examples/Image/Classification/ResNet

DonaldTsang commented 4 years ago

There are so many models to choose from

@BZbyr perhaps you can pick some examples that you would like to see.

DonaldTsang commented 4 years ago

@KichangKim if I am allowed to say this, are there any benchmarks or tests on multiple CNN architectures?

KichangKim commented 4 years ago

@KichangKim if I am allowed to say this, are there any benchmarks or tests on multiple CNN architectures?

In very early version of DeepDanbooru, I tested some types of CNN, like Inception, Squeeze-and-Excitation and so on. But not fully benchmarked it and chose ResNet for its simplicity and good reference code (in CNTK's repository). If you want, you can test yourself by extending deepdanbooru.model namespace.