presageboat / imageclassification Goto Github PK
View Code? Open in Web Editor NEWUse pytorch to implement mainstream classification networks such as resnet, mobilenetv3, regnet, resnext, efficientnet, shufflenetv2, etc. The project includes training, testing, and tensorrt inference
License: Apache License 2.0